师兄您好!有幸阅读了您的论文。想请教两个问题: 一、由于我尝试过直接使用MHSA加入C3中,然后分别将C3-MHSA模块加入到和您的网络差不多的位置,但是发现显存占用很大。而您的模块加了那么多个地方是因为每次使用都把通道数降低了的,从而可以降低显存占用是吗?如下图,head中第一个C3ResAtnMHSA模块,输入512通道,输出32...... 二、我如果直接使用您这个C3ResAtnMHSA模块将入到我的yolov5s改进的网络中,除了修改common和yolo的那两个文件以外,还有别的地方需要修改吗?我的输入图像是768大小的,就会报错。 
Activity
WindVChen commentedon Dec 15, 2023
哈喽 @SwjtuerFz ,
“如下图,head中第一个C3ResAtnMHSA模块,输入512通道,输出32......” 这里的32不是通道数哈,是特征的尺寸大小,具体可以看下yolo.py里的parse_model()以及common.py里的C3ResAtnMHSA()的代码段.
输入768图像的话,还需要对应得调整配置文件设置,具体可以参考这里#4.
SwjtuerFz commentedon Dec 15, 2023