关于C3ResAtnMHSA模块的问题

[SwjtuerFz]

师兄您好！有幸阅读了您的论文。想请教两个问题：
一、由于我尝试过直接使用MHSA加入C3中，然后分别将C3-MHSA模块加入到和您的网络差不多的位置，但是发现显存占用很大。而您的模块加了那么多个地方是因为每次使用都把通道数降低了的，从而可以降低显存占用是吗？如下图，head中第一个C3ResAtnMHSA模块，输入512通道，输出32......
二、我如果直接使用您这个C3ResAtnMHSA模块将入到我的yolov5s改进的网络中，除了修改common和yolo的那两个文件以外，还有别的地方需要修改吗？我的输入图像是768大小的，就会报错。

[WindVChen]

哈喽 @SwjtuerFz ,

“如下图，head中第一个C3ResAtnMHSA模块，输入512通道，输出32......” 这里的32不是通道数哈，是特征的尺寸大小，具体可以看下yolo.py里的parse_model()以及common.py里的C3ResAtnMHSA()的代码段.

输入768图像的话，还需要对应得调整配置文件设置，具体可以参考这里#4.

[SwjtuerFz]

好的，感谢回复！我还改了yolo文件中的另一个参数，跑通了。

…

---- 回复的原邮件 ---- | 发件人 | ***@***.***> | | 发送日期 | 2023年12月15日 17:56 | | 收件人 | WindVChen/DRENet ***@***.***> | | 抄送人 | SwjtuerFz ***@***.***>, Mention ***@***.***> | | 主题 | Re: [WindVChen/DRENet] 关于C3ResAtnMHSA模块的问题 (Issue #21) | 哈喽 @SwjtuerFz , “如下图，head中第一个C3ResAtnMHSA模块，输入512通道，输出32......” 这里的32不是通道数哈，是特征的尺寸大小，具体可以看下yolo.py里的parse_model()以及common.py里的C3ResAtnMHSA()的代码段. 输入768图像的话，还需要对应得调整配置文件设置，具体可以参考这里#4. — Reply to this email directly, view it on GitHub, or unsubscribe. You are receiving this because you were mentioned.Message ID: ***@***.***>