whai362 / PVT

Official implementation of PVT series
Apache License 2.0
1.73k stars 246 forks source link

SRA part of the paper description is inconsistent with the code implementation #74

Open torchmyheart opened 3 years ago

torchmyheart commented 3 years ago

Thank you very much to the author for sharing excellent ideas and code, but I have a little problem. In the SRA description part of the paper, in formula 3, the channel is changed by matrix multiplication after reshape first, but in the code, it is realized by a convolution. The complexity of the two is different. The author has experimented in two ways Is the result similar?Looking forward to your answer.