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能否用于yolov5和yolov4网络配置文件yaml和cfg的转换 #78

Open yao-zheng-yi opened 3 years ago

yao-zheng-yi commented 3 years ago

你好，我尝试使用yaml2cfg.py将yolov5s.yaml转换为yolov4的cfg文件，发现好像转完格式完全不一样，请问yaml2cfg.py能否用于yolov5和yolov4网络配置文件yaml和cfg的转换吗？ 或者说yolov5和yolov4网络配置文件yaml和cfg是不能相互转换的？

enazoe commented 3 years ago

只能用于v5,yolov3.4都是采用原版darknet的模型，不支持其他版本的

yao-zheng-yi commented 3 years ago

@enazoe 所以yaml2cfg.py得到的yolov5s.cfg是只能用于TensorT吗？ YOLOv5的yolov5s.yaml没办法转成像YOLOv4的yolov4.cfg那样的吗？

enazoe commented 3 years ago

是的，没办法，两种框架不同的地方太多