YOLOv5系列(3)——YOLOv5修改网络结构
最新推荐文章于 2024-08-06 14:51:21 发布
本文介绍了如何在YOLOv5中将网络结构改为MobilenetV2,并添加注意力模块。首先在models/common.py中实现MobileNetV2的bottleneck和Pointwise Convolution,接着修改yolo.py的parse_model函数以引入新模块。通过创建yolov5-mobilenetV2.yaml配置文件,成功替换backbone。此外,还展示了在yolov5x_se.yaml中添加SELayer注意力模块的步骤。
摘要由CSDN通过智能技术生成