yolov8更换bottleneck
时间: 2023-11-03 21:07:13 浏览: 116
JAVA-bottleneck.rar_bottleneck
在YOLOv8中更换主干网络的步骤如下:
1. 在ultralytics/models/v8文件夹下新建一个yolov8-ghostnet.yaml文件。
2. 将Ghostnet的代码添加到ultralytics/nn/modules/block.py文件末尾。
3. 将G_bneck这个类的名字加入到ultralytics/nn/tasks.py中。
4. 修改yolov8-ghostnet.yaml,使用G_bneck构建GhostNet主干网络。
5. 开始训练。
需要注意的是,这里提到的Bottleneck是指Bottleneck组成CSP的结构,而不是Bottleneck的内部结构。
阅读全文