mohamedac29 / S2-FPN

Scale-ware Strip Attention Guided Feature Pyramid Network for Real-time Semantic Segmentation
MIT License
21 stars 5 forks source link

SpatialAttention class #5

Open llfzllfz opened 2 years ago

llfzllfz commented 2 years ago

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.