custom pt to cfg, weight build

enazoe / yolo-tensorrt

TensorRT8.Support Yolov5n,s,m,l,x .darknet -> tensorrt. Yolov4 Yolov3 use raw darknet *.weights and *.cfg fils. If the wrapper is useful to you,please Star it.

MIT License

1.18k stars 316 forks source link

custom pt to cfg, weight build #167

Open ONNONS opened 2 years ago

ONNONS commented 2 years ago

How to convert pre-trained pt files?

Execute yaml2cfg.py to download yolov5s.pt.

What if I want to use the model I trained on?