search

faustomorales / vit-keras

Keras implementation of ViT (Vision Transformer)
Apache License 2.0
329 stars 78 forks source link

image size #24

Closed daeing closed 3 years ago

daeing commented 3 years ago

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)?

faustomorales commented 3 years ago

The library does not support non-square images. See https://github.com/faustomorales/vit-keras/issues/18.