faustomorales / vit-keras

Keras implementation of ViT (Vision Transformer)

Geek Repo:Geek Repo

Github PK Tool:Github PK Tool

image size

daeing opened this issue · comments

I found the image_size must be int. it means the picture has the same height and width? But how to handle the picture.shape = (620, 32)?

The library does not support non-square images. See #18.