S2-FPN
S2-FPN copied to clipboard
SpatialAttention class
In this class, I find mxpool = F.max_pool2d(x, [h, 1])
, and after this, maxpool will be (B, C, 1, W), but in paper, it's (B, C, H, 1). I have miss about it. Please explain, thanks.