【YOLOv5改进系列】前期回顾: YOLOv5改进系列(0)——重要性能指标与训练结果评价及分析 YOLOv5改进系列(1)——添加SE注意力机制 YOLOv5改进系列(2)——添加CBAM注意力机制 YOLOv5改进系列(3)——添加CA注意力机制 YOLOv5改进系列(4)——添加ECA注意力机制 YOLOv5改进系列(5)——替换主干网络之 MobileNetV3 YOLOv5改进系列(6)——替换主干网络之 ShuffleNetV2 目录 🚀一、SimAM介绍 1.1 简介 1.2 方法 🚀二、在backbone末端添加SimAM注意力机制方法 2.1 添加顺序 2.2 具体添加步骤 第①步:在common.py中添加SimAM模块 第②步:在yolo.py文件里的parse_model函数加入类名 第③步:创建自定义的yaml文件 第④步:验证是否加入成功 第⑤步:修改train.py中 ‘--cfg’默认参数