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.12k forks source link

How to change .pt to weight? #515

Open Barry-Chen-yup opened 1 year ago

Barry-Chen-yup commented 1 year ago

The outupts of Darknet model are .weight. And I want to change the output of pytorch model to .weight. How can I do?

WongKinYiu commented 1 year ago

yolov7.weights

Barry-Chen-yup commented 1 year ago

I mean the trained model output weight, not pretrain weight.

developer239 commented 1 year ago

@WongKinYiu Thanks for the link to the pre-trained weights. I just used your repo to train my own model. It worked great. I know that I can export to .onnx and that I can use onnx runtime but I was also wondering how do I convert .pt file to .weights so that I can use DNN and OpenCV.

by011 commented 1 year ago

yolov7.weights

sorry, can you please send the .config file as well ?

WongKinYiu commented 1 year ago

https://github.com/AlexeyAB/darknet/blob/master/cfg/yolov7.cfg

by011 commented 1 year ago

https://github.com/AlexeyAB/darknet/blob/master/cfg/yolov7.cfg

thank you very much

developer239 commented 1 year ago

@by011 https://github.com/developer239/yolo-cmake-opencv-onnx-cpp