yolov5更换主干网络fasternet
时间: 2023-09-10 07:12:45 浏览: 208
在YOLOv5中更换主干网络为Fasternet,需要进行以下步骤:
1. 在ultralytics/models/v8文件夹下新建一个名为yolov8-FasterNet.yaml的文件。
2. 将FasterNet的代码添加到ultralytics/nn/modules.py文件的末尾。
3. 将FasterNet的类名添加到ultralytics/nn/tasks.py中。
4. 修改yolov8-FasterNet.yaml文件,使用PatchEmbed_FasterNet、BasicStage和PatchMerging_FasterNet构建Fasternet主干网络。
5. 修改ultralytics/yolo/cfg/default.yaml文件的'--model'默认参数,或者使用指令直接开始训练。
综上所述,以上是在YOLOv5中更换主干网络为Fasternet所需的步骤。<span class="em">1</span><span class="em">2</span><span class="em">3</span>
#### 引用[.reference_title]
- *1* *3* [主干网络篇 | YOLOv8 更换主干网络之 FasterNet | 《Ran, Don‘t Walk: 追求更高 FLOPS 的快速神经网络》](https://blog.csdn.net/weixin_43694096/article/details/130291796)[target="_blank" data-report-click={"spm":"1018.2226.3001.9630","extra":{"utm_source":"vip_chatgpt_common_search_pc_result","utm_medium":"distribute.pc_search_result.none-task-cask-2~all~insert_cask~default-1-null.142^v93^chatsearchT3_2"}}] [.reference_item style="max-width: 50%"]
- *2* [YOLOV5的多主干网络(backbone)实现.zip](https://download.csdn.net/download/cuihao1995/83477139)[target="_blank" data-report-click={"spm":"1018.2226.3001.9630","extra":{"utm_source":"vip_chatgpt_common_search_pc_result","utm_medium":"distribute.pc_search_result.none-task-cask-2~all~insert_cask~default-1-null.142^v93^chatsearchT3_2"}}] [.reference_item style="max-width: 50%"]
[ .reference_list ]
阅读全文