faustomorales / vit-keras

Keras implementation of ViT (Vision Transformer)

Geek Repo:Geek Repo

Github PK Tool:Github PK Tool

Support of non-square size image

kelvinkoh0308 opened this issue · comments

Hi,

Currently, the vit-keras only support square size input ( 256x256), could you change it to support non-square size input image like 256 x 128?

Thank you very much.

Supporting non-square input sizes would be non-trivial because it changes how we load pre-trained weights and has ramifications for patch sizes. A PR with support on this would be welcome but I'm afraid I cannot devote time to this in the foreseeable future.

Closing since this is not a bug. A PR adding support for non-square images would be welcome.

Non-square functionality added by #26.