how to train yolov4-csp on yolov4-large branch?

WongKinYiu / ScaledYOLOv4

Scaled-YOLOv4: Scaling Cross Stage Partial Network

GNU General Public License v3.0

2.02k stars 570 forks source link

how to train yolov4-csp on yolov4-large branch? #151

Open LukeAI opened 3 years ago

LukeAI commented 3 years ago

I managed to get training yolov4-csp from the provided .weights working on the yolov4-csp branch https://github.com/WongKinYiu/ScaledYOLOv4/issues/125 but in the yolov4-large branch there is no load_darknet_weights function and the data structures appear to be a little different. Would it be possible to provide yolov4-csp.pt ?

WongKinYiu commented 3 years ago

.pt and use the hyp*.yaml in yolov4-csp branch for training.

LukeAI commented 3 years ago

thanks so much!

I see that on the yolov4-csp branch https://github.com/WongKinYiu/ScaledYOLOv4/tree/yolov4-csp/data there is only hyp.scratch.yaml - no hyp.finetune.yaml - do you have recommended hyp.finetune.yaml also?

qutyyds commented 2 years ago

这是yolov4-csp.pt吗？