yolov7中Concat之后加注意力模块(最复杂的情况)

yolov7中Concat之后加注意力模块(最复杂的情况)_第1张图片
1、common.py中找到Concat模块,复制一份
yolov7中Concat之后加注意力模块(最复杂的情况)_第2张图片
2、要传参进来,dim通道数
yolov7中Concat之后加注意力模块(最复杂的情况)_第3张图片
3、然后找yolo.py模块,添加
yolov7中Concat之后加注意力模块(最复杂的情况)_第4张图片
4、yaml里替换
yolov7中Concat之后加注意力模块(最复杂的情况)_第5张图片
5、和加的位置也有关系

你可能感兴趣的:(yolov7,YOLO)