No detection with custom weight

[ndmreda]

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]

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]

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

[selfcar2022]

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