yolov4-deepsort
yolov4-deepsort copied to clipboard
No detection with custom weight
Hello, I tried to run the algorithm on my custom weights file but when I launch the code on my camera it detects nothing knowing that I tested the saved_model.pb file and it detects well, Did anyone come across any problems that I have and came up with a solution, please? I use Tensorflow-gpu 2.6
try replacing save_model.py in the current repository by save_model.py in theAIGuysCode/yolov4-custom-functions then save the model again and don't forget to change the model config and classes file
I found that there are some issue in saving the weights when you use GPU to run the code. hope it will fix your problem.
it doesn't work, I think it's the same file there is no difference between the two
Did you reinstall tensorflow-gpu==2.3.0 ?
Installing that version of Tensorflow solved the Binding Box problem.
So try the next code.
pip install tensorflow-gpu==2.3.0