search

WongKinYiu / yolov7

Implementation of paper - YOLOv7: Trainable bag-of-freebies sets new state-of-the-art for real-time object detectors
GNU General Public License v3.0
13.01k stars 4.11k forks source link

How can I change the pytorch weights (.pt) to darknet weights (.weights)? #1967

Open DongNaeSwellfish opened 5 months ago

DongNaeSwellfish commented 5 months ago

I want to use darknet for yolov7 model that is trained with my custom data on this github code. However, I cannot find any way to convert the pytorch weights to darknet weight format. Can anyone please tell me whether it is possible or not?

dsbyprateekg commented 5 months ago

I want to use darknet for yolov7 model that is trained with my custom data on this github code. However, I cannot find any way to convert the pytorch weights to darknet weight format. Can anyone please tell me whether it is possible or not?

Darknet's version of yolov7 will not give you better results as compared to their yolov4. I suggest you use darknet's yolov4 if you want to stick to darknet.

ArthurTlprt commented 1 week ago

I'm also interested in finding a way to convert the weight file, containing the model trained using this repo into darknet .weights.

@DongNaeSwellfish, were you able to find a solution?