problem with videos - Githubissues

hunglc007 / tensorflow-yolov4-tflite

YOLOv4, YOLOv4-tiny, YOLOv3, YOLOv3-tiny Implemented in Tensorflow 2.0, Android. Convert YOLO v4 .weights tensorflow, tensorrt and tflite

https://github.com/hunglc007/tensorflow-yolov4-tflite

MIT License

2.23k stars 1.24k forks source link

problem with videos #356

Open khoatran94 opened 3 years ago

khoatran94 commented 3 years ago

The object detection is done only with the first frame of the video. Please help

Noobzik commented 3 years ago

Same here, first frame only.

Seems like infer(batch_data) run only once

Pedrohgv commented 3 years ago

I think I found a way to solve it. I was using a tensorflow 2.4.2 docker image, changing to the tensorflow version specified in the requirements-gpu.txt file (2.3.0rc0-gpu) solved it, ran on GPU, got predictions for the entire video, and fast as usual. I tested predicting on a model converted using the 2.4.2 version and it didnt't work even with predicting with the 2.3.0rc0 version. I had to re-convert the model using TF 2.3.0rc0, and also predict with 2.3.0rc0.

Clark1216 commented 2 years ago

But RTX30 cannot use 2.3.0