问题遇到的现象和发生背景
想再yolov5网络模块中添加CA注意力机制模块
yolo.py可以跑通,但train.py的计算map过程中出现特征尺度不匹配,希望得到[32, 128, 168 168]却得到[32, 128, 112, 168],改变bitchsize时得到[32, 128, 128, 168]
问题相关代码,请勿粘贴截图
下为train.py报错图片
下为网络模块添加图
下为网络结构配置文件
想再yolov5网络模块中添加CA注意力机制模块
yolo.py可以跑通,但train.py的计算map过程中出现特征尺度不匹配,希望得到[32, 128, 168 168]却得到[32, 128, 112, 168],改变bitchsize时得到[32, 128, 128, 168]
下为train.py报错图片