Bounding boxes not showing when using `tensorflow-gpu`

Question

Bounding boxes not showing when using `tensorflow-gpu`

yassine-rd opened this issue 2 years ago · 10 comments

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 a tensorflow-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?

Answer 1 · 2022-12-01T00:25:49.000Z

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.
However, when I run the same command using only cpu, the bounding boxes and ids from the output of object_tracker.py are shown.

Please fix it or let me know how to get the bounding boxes and ids.

Answer 2 · 2022-12-01T04:31:11.000Z

I'm using Google Collab with GPU and this requirements-gpu.txt file changes, and it's working.
Still trying to fix to run it with my PC macOS M1 Pro with CPU.

tensorflow-gpu==2.3.0
opencv-python==4.1.2.30
lxml
tqdm
absl-py
matplotlib
easydict
pillow
numpy==1.18.5
scipy==1.4.1

Answer 3 · 2022-12-01T07:58:01.000Z

I'm using Google Collab with GPU and this requirements-gpu.txt file changes, and it's working. Still trying to fix to run it with my PC macOS M1 Pro with CPU.
tensorflow-gpu==2.3.0
opencv-python==4.1.2.30
lxml
tqdm
absl-py
matplotlib
easydict
pillow
numpy==1.18.5
scipy==1.4.1

Are you able to run your object_detector.py directly on your machine using a GPU env and have bounding boxes shown @MaximilianoAdaro?
PS : I'm using a macOS M1 too

Answer 4 · 2022-12-01T21:10:07.000Z

Thanks!
I did well.

Answer 5 · 2023-02-01T21:42:09.000Z

Thanks! I did well.

Hi, how did you end up resolving this?

Answer 6 · 2023-02-02T02:47:21.000Z

No problem.

Answer 7 · 2023-03-13T07:38:51.000Z

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. However, when I run the same command using only cpu, the bounding boxes and ids from the output of object_tracker.py are shown.

Please fix it or let me know how to get the bounding boxes and ids.

too

How did you end up resolving this problem @Ishihara-Masabumi ?
Let me know how to get the bounding boxes and ids. I has same issue now

Answer 8 · 2023-03-13T08:34:21.000Z

OK, thanks.

Answer 9 · 2023-04-17T07:37:44.000Z

I'm having the same issue, any suggestions for solution?
I tried to run Yolo4 separately on darknet and it's working fine but when I use the same .weight file with deepsort it doesn't work, any suggestions?

Answer 10 · 2023-04-20T12:31:38.000Z

I fixed it by using commit it pull request! PLEASE CHECK IT AND INFORM IF IT STILL HAS ISSUES.