I’ve been training a YOLOv4 object detection model with TAO, but the results are not satisfactory. The highest mAP value I’ve achieved is around 0.455, which is quite low compared to the results I get when training with Darknet. In Darknet, my initial mAP value is around 0.64. My highest mAP is over 80. I’m trying to understand the problem causing this difference in performance.
Has anyone else experienced similar issues when training YOLOv4 with TAO compared to Darknet? Any suggestions on how to improve the mAP values when training with TAO?
TAO config:
yolov4_train_resnet101.txt (2.2 KB)
TAO training log:
training_log.csv (2.5 KB)
Darknet config:
yolov4-custom.cfg (12.0 KB)