yoloV8主干网络
时间: 2023-09-29 12:09:26 浏览: 207
yolov8项目文件2023/3/18版
YOLOv8主干网络可以更换为ShuffleNetv2。要进行这个更换,可以按照以下步骤进行操作:在ultralytics/models/v8文件夹下新建一个yolov8-ShuffleNetv2.yaml文件;将提供的ShuffleNetv2代码添加到ultralytics/nn/modules.py文件的末尾;将ShuffleNetv2类的名称添加到ultralytics/nn/tasks.py中;修改yolov8-ShuffleNetv2.yaml文件,使用Shuffle_Block构建ShuffleNetv2主干网络;修改ultralytics/yolo/cfg/default.yaml文件的'--model'默认参数或直接使用相应指令开始训练。
阅读全文