Views
No views yet
facebook/dino-vits16 for zeromodels. One implementation runs unmodified on TensorFlow / Torch / JAX.DinoViTModel), not a task head.1import os
2os.environ["KERAS_BACKEND"] = "torch" # or "jax" / "tensorflow"
3
4from zeromodels.models.dino import DinoViTModel, DinoImageProcessor
5
6# The processor resizes + ImageNet-normalizes, so build the model with
7# include_normalization=False (it would otherwise normalize a second time).
8model = DinoViTModel.from_weights(
9 "zeromodels/dino-vits16", include_normalization=False
10)
11processor = DinoImageProcessor.from_weights("zeromodels/dino-vits16")
12
13pixel_values = processor("your_image.jpg")["pixel_values"]
14features = model(pixel_values, training=False)
15print(pixel_values.shape, features.shape)from_weights("zeromodels/<variant>"):| Variant | Hub | Backbone |
|---|---|---|
dino-vits16 | zeromodels/dino-vits16 | ViT-S/16 |
dino-vits8 | zeromodels/dino-vits8 | ViT-S/8 |
dino-vitb16 | zeromodels/dino-vitb16 | ViT-B/16 |
dino-vitb8 | zeromodels/dino-vitb8 | ViT-B/8 |
dino-resnet50 | zeromodels/dino-resnet50 | ResNet-50 |
KERAS_BACKEND before importing Keras / zeromodels.include_normalization=False. To skip it, feed raw [0, 255] pixels and keep the default include_normalization=True.dino-resnet50 was converted from torch.hub facebookresearch/dino.DinoViTModel.from_weights("hf:facebook/dino-vits16").