keras-yolo3 icon indicating copy to clipboard operation
keras-yolo3 copied to clipboard

low fps and no bounding box on Tensorflow 2

Open sachinkmohan opened this issue 2 years ago • 5 comments

I was testing this repository https://github.com/susnato/keras-yolo3 by @susnato on Tensorflow 2.6.0

But when I run the below

python yolo_video.py --model model_data/yolo.h5  --input /home/mohan/git/backups/drive.mp4 --anchors model_data/yolo_anchors.txt --classes model_data/coco_classes.txt

I am getting low fps and no bounding boxes are generated too. Can someone please help.

I was getting the below error when running the inference before

ValueError: Subshape must have computed start >= end since stride is negative, but is 0 and 2 

So I applied the fix given here -> https://stackoverflow.com/questions/57558476/training-a-keras-model-yields-multiple-optimizer-errors

Now my output looks like the below.

https://user-images.githubusercontent.com/26414662/165150438-2456eee1-0d76-4fe7-9c50-4de7283813d3.mp4

sachinkmohan avatar Apr 25 '22 18:04 sachinkmohan

Hi @sachinkmohan , can you please provide some additional information about how to reproduce this issue? Any Colab notebook or something would be useful. Actually I created TF 2.6 version of this repo a long time ago and don't remember the steps but will look for the cause of this issue and try to fix it.

susnato avatar Apr 25 '22 18:04 susnato

Thanks @susnato for writing back! :)

I made a fork of your repository -> https://github.com/sachinkmohan/keras-yolo3 Make sure you checkout the tf_2.x_compatible branch to see the updated README file and commits.

Once you have everything in place just execute the below. Please make sure you convert and generate yolo.h5 file as given in the old README.

python yolo_video.py --model model_data/yolo.h5  --input /home/mohan/git/backups/drive.mp4 --anchors model_data/yolo_anchors.txt --classes model_data/coco_classes.txt

sachinkmohan avatar Apr 26 '22 10:04 sachinkmohan

Did anyone solve this issue? I have the same issue. The prediction results look like it correctly found objects, but every bounding box is located left-top of the image.

ksk0629 avatar Sep 27 '22 01:09 ksk0629

Same problem, looking for the latest update of this issue

hv0905 avatar Nov 19 '22 13:11 hv0905

I figured out the problem was due to the shape.

ksk0629 avatar Nov 19 '22 13:11 ksk0629