Hugging Face's logo

mlx-vision
/
vit_base_patch16_384.swag_e2e-mlxim

Image Classification
mlx-image
Safetensors
MLX
vision
Model card Files Community
riccardomusmeci commited on
Commit
62d9f3a
•
1 Parent(s): bfa5ced

Update README.md

Browse files
Files changed (1) hide show
  1. README.md +2 -2
README.md CHANGED
@@ -28,7 +28,7 @@ from mlxim.model import create_model
28
  from mlxim.io import read_rgb
29
  from mlxim.transform import ImageNetTransform
30
 
31
- transform = ImageNetTransform(train=False, img_size=224)
32
  x = transform(read_rgb("cat.png"))
33
  x = mx.expand_dims(x, 0)
34
 
@@ -44,7 +44,7 @@ from mlxim.model import create_model
44
  from mlxim.io import read_rgb
45
  from mlxim.transform import ImageNetTransform
46
 
47
- transform = ImageNetTransform(train=False, img_size=224)
48
  x = transform(read_rgb("cat.png"))
49
  x = mx.expand_dims(x, 0)
50
 
 
28
  from mlxim.io import read_rgb
29
  from mlxim.transform import ImageNetTransform
30
 
31
+ transform = ImageNetTransform(train=False, img_size=384)
32
  x = transform(read_rgb("cat.png"))
33
  x = mx.expand_dims(x, 0)
34
 
 
44
  from mlxim.io import read_rgb
45
  from mlxim.transform import ImageNetTransform
46
 
47
+ transform = ImageNetTransform(train=False, img_size=384)
48
  x = transform(read_rgb("cat.png"))
49
  x = mx.expand_dims(x, 0)
50