search

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 313 forks source link

关于yaml2cfg.py #48

Closed lq0104 closed 3 years ago

lq0104 commented 3 years ago

您好，您提供的yaml2cfg.py转换后的cfg和weights模型是不是只能用于tensorRT的推理，是不能被AlexeyAB darknet的repo所训练的对吗

enazoe commented 3 years ago

@lq0104 是的，只是一种组织方式而已，是为trt服务的

lq0104 commented 3 years ago

好的，明白了

enazoe commented 3 years ago

@lq0104 觉得能用的话，star下哈

lq0104 commented 3 years ago

好的，已Star，如果您的yaml2cfg脚本能够支持darknet训练可能对于想使用darknet进行finetune，darknet和yolov5训练模型MAP/速度的比对以及终端部署更方便了哈 :)