Views
No views yet
facebook/dinov2-small for zeromodels. One implementation runs unmodified on TensorFlow / Torch / JAX.DinoV2Model), not a task head.1import os
2os.environ["KERAS_BACKEND"] = "torch" # or "jax" / "tensorflow"
3
4from zeromodels.models.dino_v2 import DinoV2Model, DinoV2ImageProcessor
5
6# The processor resizes + ImageNet-normalizes, so build the model with
7# include_normalization=False (it would otherwise normalize a second time).
8model = DinoV2Model.from_weights(
9 "zeromodels/dinov2-small", include_normalization=False
10)
11processor = DinoV2ImageProcessor.from_weights("zeromodels/dinov2-small")
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 |
|---|---|---|
dinov2-small | zeromodels/dinov2-small | ViT-S/14 |
dinov2-base | zeromodels/dinov2-base | ViT-B/14 |
dinov2-large | zeromodels/dinov2-large | ViT-L/14 |
dinov2-giant | zeromodels/dinov2-giant | ViT-g/14 |
KERAS_BACKEND before importing Keras / zeromodels.include_normalization=False. To skip it, feed raw [0, 255] pixels and keep the default include_normalization=True.DinoV2Model.from_weights("hf:facebook/dinov2-small").