前言:因为本人是在初次修改模型(yolov9),所以想做个总结方便自己查看和给第一次修改模型没有方向的人一些简单的建议(如有不对请多多给出建议)(未完成后续还会完善)
模型基本由backbone,neck和检测头组成,以yolov9为例,其中绿色框为backbone,蓝色框为neck,检测头为DualDDetect
1.backbone修改
1.1整体替换backbone
把backbone整体替换为其他网络如VGG、ResNet、MobileNet、EfficientNet等
1.2部分替换backbone
往backbone骨干网络添加注意力,或者对特点卷积进行更换
2.neck修改
2.1整体替换neck
2.2部分替换neck
往backbone骨干网络添加注意力(感觉往backbone中添加注意力效果更好),或者对特点卷积进行更换
3.检测头修改
4.改进损失函数
5.剪枝蒸馏
6.注意点:
消融实验
消融实验是一种常见的机器学习实验方法,用于评估特定组件、特征或方法对整体模型性能的影响。说白了就是A对模型好,B对模型好,A+B不一定对模型好,然后要单独去看(必须做)
注意力机制
发论文,创新性>提点
项目,提点>创新行
发论文,对于模块添加注意力机制,除非是新的(近两年CVPR),否则虽然能提点但是创新新不强,但是可以作为部分工作,来增加工作量