yolov5yaml
时间: 2023-10-10 20:05:47 浏览: 148
yolov5
yolov5的网络模型结构由位于models文件夹下的yaml文件定义。例如,yolov5的6.0版本中,models文件夹下有多个yaml文件,包括yolov5n.yaml、yolov5s.yaml、yolov5m.yaml、yolov5l.yaml和yolov5x.yaml。这些yaml文件的区别仅在于depth_multiple和width_multiple这两个参数的不同。尽管这些文件中backbone和head部分完全相同,但通过调整depth_multiple和width_multiple这两个参数,可以实现不同复杂度的模型设计。
重新进行训练时,可以通过设置train.py中的weights参数为yolov5_change.yaml文件来使用修改后的模型。在控制台输出信息中,可以看到相应的提示,表示修改成功。
如果想查看模型的详细结构,可以使用yolov5目录下的export.py文件将.pt文件转换为.onnx文件。
阅读全文