1、ymal文件修改
将models文件下yolov5s.py复制重命名如下图所示:
2、接着将如下代码替换,diamagnetic如下所示:
# YOLOv5 🚀 by Ultralytics, GPL-3.0 license
# Parameters
nc: 1 #
本文介绍了如何将YOLOV5的骨干网络替换为MobileNetV3,涉及ymal、common.py、yolo.py和train文件的修改,通过这些步骤实现模型的轻量化。在训练过程中,详细说明了每个文件的改动,并展示了运行训练的截图。欢迎读者指出可能存在的错误。
将models文件下yolov5s.py复制重命名如下图所示:
2、接着将如下代码替换,diamagnetic如下所示:
# YOLOv5 🚀 by Ultralytics, GPL-3.0 license
# Parameters
nc: 1 #
1579
967
6153

被折叠的 条评论
为什么被折叠?