Image size other than default ones doesn't work

rishigami / Swin-Transformer-TF

Tensorflow implementation of Swin Transformer model.

Apache License 2.0

204 stars 50 forks source link

Closed awsaf49 closed 3 years ago

awsaf49 commented 3 years ago

Notebook: https://colab.research.google.com/drive/1nqYkQCUzShkVdqGxW4TyMrtAb0n5MBZR#scrollTo=G9ZVlphmqD7d Issue:
In swin_tiny_224 I've tried multiple of 224, 512x512, multiple of window_size. But nothing seems to work other than the 224x224.
Same goes for swin_large_384, only default size 384x384 works.

I'm wondering if this is expected behavior or not. Is there any way to make it work for non-square image?

awsaf49 commented 3 years ago

So, it seems that SwinTransformer is bound to 384 or 224.