Yolo-Pytorch Train and use YOLO with pytorch Please modify utils/datasets.py line 53 to correct image extension (jpg/jpeg/png) For config/cfg file: filters = ([Number of classes] + 5) x 3