We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.
By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.
Already on GitHub? Sign in to your account
师兄您好!有幸阅读了您的论文。想请教两个问题: 一、由于我尝试过直接使用MHSA加入C3中,然后分别将C3-MHSA模块加入到和您的网络差不多的位置,但是发现显存占用很大。而您的模块加了那么多个地方是因为每次使用都把通道数降低了的,从而可以降低显存占用是吗?如下图,head中第一个C3ResAtnMHSA模块,输入512通道,输出32...... 二、我如果直接使用您这个C3ResAtnMHSA模块将入到我的yolov5s改进的网络中,除了修改common和yolo的那两个文件以外,还有别的地方需要修改吗?我的输入图像是768大小的,就会报错。
The text was updated successfully, but these errors were encountered:
哈喽 @SwjtuerFz ,
“如下图,head中第一个C3ResAtnMHSA模块,输入512通道,输出32......” 这里的32不是通道数哈,是特征的尺寸大小,具体可以看下yolo.py里的parse_model()以及common.py里的C3ResAtnMHSA()的代码段.
输入768图像的话,还需要对应得调整配置文件设置,具体可以参考这里#4.
Sorry, something went wrong.
No branches or pull requests
师兄您好!有幸阅读了您的论文。想请教两个问题:
一、由于我尝试过直接使用MHSA加入C3中,然后分别将C3-MHSA模块加入到和您的网络差不多的位置,但是发现显存占用很大。而您的模块加了那么多个地方是因为每次使用都把通道数降低了的,从而可以降低显存占用是吗?如下图,head中第一个C3ResAtnMHSA模块,输入512通道,输出32......
二、我如果直接使用您这个C3ResAtnMHSA模块将入到我的yolov5s改进的网络中,除了修改common和yolo的那两个文件以外,还有别的地方需要修改吗?我的输入图像是768大小的,就会报错。
The text was updated successfully, but these errors were encountered: