ZeroDivisionError when trying to train yolo_v4 with TLT3

Accelerated Computing Intelligent Video Analytics Transfer Learning Toolkit
#1

Hello
I try to train a yolo_v4 model, and when I try to train I get this error:

ZeroDivisionError: float division by zero

Thanks

#2

I just noticed that before the error I get this message:

2021-04-13 09:27:07,950 [INFO] main: Number of images in the training dataset: 0

I checked all of the paths and they all seem to be fine.

#3

I found the problem.
all the image files had “.JPEG” instead of “.jpeg” suffix. (uppercase instead of lowercase).

1 Like
#4

Glad to know issue resolved, thanks for the update.