Transfer Learning: Train on Traffic Signs with pre-trained weights

[ridwan102]

I'm attempting to train my Yolo object detector using the Darknet CNN. I'm using Yolov4 pre-trained weights which as you showed can predict Cars, Traffic Lights, and Stop Signs with these COCO Classes. Just wondering how I can add an extra layer so my model can also pick up Traffic Signs. Code below to train:

!./darknet detector train data/obj.data cfg/yolov4-obj.cfg yolov4.conv.137 -dont_show -map