yolov8更改主干
时间: 2023-10-31 14:07:38 浏览: 278
可更改骨干网络添加注意力yolov8-的开发版
Yolov8更改主干网络的步骤如下:
1. 在ultralytics/models/v8文件夹下新建一个yolov8-新主干网络名称.yaml文件。
2. 将新主干网络的代码添加到ultralytics/nn/modules/block.py文件末尾。
3. 将CBRM和新主干网络名称添加到ultralytics/nn/tasks.py中。
4. 修改yolov8-新主干网络名称.yaml文件,使用CBRM和新主干网络构建新的主干网络 。
5. 在训练时,使用修改过后的yolov8-新主干网络名称.yaml配置文件替换原有的配置文件。
--相关问题--:
阅读全文