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
PSA下面这段代码里面的channel和reduction分别是什么意思,本人是新手,望大佬指导一下,不知道如何设定成自己想要的 from model.attention.PSA import PSA import torch input=torch.randn(50,512,7,7) psa = PSA(channel=512,reduction=8) output=psa(input) print(output.shape)
The text was updated successfully, but these errors were encountered:
channel=512就是输入通道,根据你的输入特征图改变。reduction=8应该是表示对输入特征张量的通道数压缩的程度,可以自己调看什么值对你好。
Sorry, something went wrong.
No branches or pull requests
PSA下面这段代码里面的channel和reduction分别是什么意思,本人是新手,望大佬指导一下,不知道如何设定成自己想要的
from model.attention.PSA import PSA
import torch
input=torch.randn(50,512,7,7)
psa = PSA(channel=512,reduction=8)
output=psa(input)
print(output.shape)
The text was updated successfully, but these errors were encountered: