search

Sense-X / UniFormer

[ICLR2022] official implementation of UniFormer
Apache License 2.0
812 stars 111 forks source link

SABlock is the same as SABlock_windows？ #116

Closed mywebinfo65536 closed 1 year ago

mywebinfo65536 commented 1 year ago

SABlock_Windows （1）SABlock_Windows structure SABlock （2）SABlock structure

I found them the same, why ?

Andy1621 commented 1 year ago

Yes, the structure is the same (thus the pretrained parameter can be loaded), but the operation is different. Please check https://github.com/Sense-X/UniFormer/blob/f92e423f7360b0026b83362311a4d85e448264d7/object_detection/mmdet/models/backbones/uniformer.py#L192-L210

The window block is used to reduce the computation for large-resolution input.

mywebinfo65536 commented 1 year ago

Okey, thanks for your answer.