-
Notifications
You must be signed in to change notification settings - Fork 752
New issue
Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.
By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.
Already on GitHub? Sign in to your account
Bounding boxes not showing when using tensorflow-gpu
#155
Comments
Me too, On the Linux machine of ubuntu 20.04 and Python 3.8.10 with GPU, I didn't see any bounding boxes and ids from the output of object_tracker.py. Please fix it or let me know how to get the bounding boxes and ids. |
I'm using Google Collab with GPU and this requirements-gpu.txt file changes, and it's working.
|
Are you able to run your |
tensorflow-gpu
tensorflow-gpu
on macOS
tensorflow-gpu
on macOStensorflow-gpu
Thanks! |
Hi, how did you end up resolving this? |
No problem. |
too How did you end up resolving this problem @Ishihara-Masabumi ? |
OK, thanks. |
I'm having the same issue, any suggestions for solution? |
I fixed it by using commit it pull request! PLEASE CHECK IT AND INFORM IF IT STILL HAS ISSUES. |
Hi,
So I've trained a YOLOv4 custom object detector on Colab and then downloaded the yolov4-custom.weights file.
When I run
python save_model.py --weights ./data/yolov4-custom.weights --output ./checkpoints/yolov4-custom --model yolov4
using atensorflow-gpu
env, the ./checkpoints/yolov4-custom/ has the .pb files.But when I run
python object_tracker.py --video ./data/video/test.mp4 --output ./outputs/result.avi --weights ./checkpoints/yolov4-custom --model yolov4
, I don't see any bounding boxes on the video.However, when I run the same command using a
tensorflow-cpu
env, the bounding boxes are drawn and everything works perfectly.Am I missing something?
The text was updated successfully, but these errors were encountered: