search

AlexeyAB / darknet

YOLOv4 / Scaled-YOLOv4 / YOLO - Neural Networks for Object Detection (Windows and Linux version of Darknet )
http://pjreddie.com/darknet/
Other
21.63k stars 7.95k forks source link

YOLOv4 backbone #5430

Open david8862 opened 4 years ago

david8862 commented 4 years ago

Hi @AlexeyAB , many thanks for the awesome YOLOv4 work! Just want to ask where can I get the .cfg and weights file for backbone part of the official YOLOv4 model structure? That would be useful for transfer training to other datasets

AlexeyAB commented 4 years ago

https://github.com/AlexeyAB/darknet/blob/master/cfg/csdarknet53-omega.cfg

david8862 commented 4 years ago

thanks a lot. And how about the weights file? I tried https://github.com/AlexeyAB/darknet/releases/download/darknet_yolo_v3_optimal/yolov4.conv.137 but seems they doesn't match

adonishong commented 4 years ago

thanks a lot. And how about the weights file? I tried https://github.com/AlexeyAB/darknet/releases/download/darknet_yolo_v3_optimal/yolov4.conv.137 but seems they doesn't match

I guess this link has the answer, https://github.com/AlexeyAB/darknet/wiki/Train-Detector-on-MS-COCO-(trainvalno5k-2014)-dataset

csdarknet53-omega.conv.105 - The first 105 layers from the pre-trained weights on ImageNet (ILSVRC2012):