search

theAIGuysCode / yolov4-deepsort

Object tracking implemented with YOLOv4, DeepSort, and TensorFlow.
GNU General Public License v3.0
1.31k stars 745 forks source link

No detection with custom weight #140

Open ndmreda opened 2 years ago

ndmreda commented 2 years ago

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

AliAlmuhaysin commented 2 years ago

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.

ndmreda commented 2 years ago

it doesn't work, I think it's the same file there is no difference between the two

selfcar2022 commented 2 years ago

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