ZeroDivisionError when trying to train yolo_v4 with TLT3

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

ZeroDivisionError: float division by zero


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.

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

1 Like

Glad to know issue resolved, thanks for the update.