S2-FPN icon indicating copy to clipboard operation
S2-FPN copied to clipboard

the description of SSAM is inconsistent with the code

Open jiayu-Zou opened this issue 1 year ago • 0 comments

theoretical code: att = torch.softmax(torch.mul(mxpool, avgpool), 1) fusion = attt * avgpool + attt * mxpool

actual code : att = torch.softmax(torch.mul(mxpool, avgpool), 1) attt1 = att[:, 0, :, :].unsqueeze(1) attt2 = att[:, 1, :, :].unsqueeze(1) fusion = attt1 * avgpool + attt2 * mxpool

why?

jiayu-Zou avatar Oct 14 '23 09:10 jiayu-Zou