search

faustomorales / vit-keras

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

Update visualize.py #39

Closed namdn73 closed 1 year ago

namdn73 commented 1 year ago

fix the size of image. img_height, img_width = model.input_shape[1], model.input_shape[2]