theAIGuysCode / yolov4-custom-functions

A Wide Range of Custom Functions for YOLOv4, YOLOv4-tiny, YOLOv3, and YOLOv3-tiny Implemented in TensorFlow, TFLite, and TensorRT.
MIT License
601 stars 371 forks source link

Infer in detect.py outputs zero size matrix for all images except first #48

Open navidnadery opened 3 years ago

navidnadery commented 3 years ago

I have trained yolov4-640 on the plate detection, exactly like what is said in tutorial. I used save_model.py exactly like what is said for the pre-trained one. I pass some images to detect.py, the first image is ok and the plate is detected but for the next images output of infer is zero size. If I change the sort of images and pass them again, again the first image is ok but the next one are zero size

Madhuri-Chada commented 1 year ago

Did you find the solution for this? Im facing the same error . Pleaase help me . Thanks in advance