yolov7結(jié)構(gòu)圖
方法:直接在common里改,在相關(guān)的后面加上就行
1、接受通道數(shù)的注意力機(jī)制
1、目的:在三個(gè)輸出地方添加注意力
yolov7.yaml文件,換成其他模塊
注意力鏈接
2、models下建SE.py
3、common.py下,先找class Conv,再復(fù)制一份修改,把模塊導(dǎo)進(jìn)來
4、yolo.py文件,導(dǎo)入進(jìn)來,添加
5、yolov7.yaml,修改模塊名字
6、全都變成conv_ATT,倒時(shí)候換的時(shí)候只用換common.py的這里
7、還有第三個(gè)地方?jīng)]加
8、common.py找到sppcspc,復(fù)制一份在下面
9、yolo.py添加名字
2、不接受通道的注意力機(jī)制
比如說simAm
1、models下建立文件夾,放進(jìn)去
2、common.py導(dǎo)入模塊名
3、common.py下改,默認(rèn)的可以不用動文章來源:http://www.zghlxwxcb.cn/news/detail-698976.html
3、加在連接之后
四個(gè)連接之后加
1、common.py找到Concat類,復(fù)制一份,加一個(gè)參數(shù)channel
不用接受通道數(shù)的可以不加channel參數(shù)
2、yolo.py,加這些文章來源地址http://www.zghlxwxcb.cn/news/detail-698976.html
到了這里,關(guān)于yolov7添加注意力機(jī)制的文章就介紹完了。如果您還想了解更多內(nèi)容,請?jiān)谟疑辖撬阉鱐OY模板網(wǎng)以前的文章或繼續(xù)瀏覽下面的相關(guān)文章,希望大家以后多多支持TOY模板網(wǎng)!