Skip to content

关于C3ResAtnMHSA模块的问题 #21

@SwjtuerFz

Description

@SwjtuerFz

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

Activity

WindVChen

WindVChen commented on Dec 15, 2023

@WindVChen
Owner

哈喽 @SwjtuerFz ,

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

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

SwjtuerFz

SwjtuerFz commented on Dec 15, 2023

@SwjtuerFz
Author
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment

Metadata

Metadata

Assignees

No one assigned

    Labels

    No labels
    No labels

    Projects

    No projects

    Milestone

    No milestone

    Relationships

    None yet

      Development

      No branches or pull requests

        Participants

        @WindVChen@SwjtuerFz

        Issue actions

          关于C3ResAtnMHSA模块的问题 · Issue #21 · WindVChen/DRENet