KerasHub
prasadsachin commited on
Commit
f524393
·
verified ·
1 Parent(s): 196e39b

Update README.md with new model card content

Browse files
Files changed (1) hide show
  1. README.md +46 -20
README.md CHANGED
@@ -1,23 +1,49 @@
1
  ---
2
  library_name: keras-hub
3
  ---
4
- This is a [`DINOV2` model](https://keras.io/api/keras_hub/models/dinov2) uploaded using the KerasHub library and can be used with JAX, TensorFlow, and PyTorch backends.
5
- Model config:
6
- * **name:** dinov2_backbone
7
- * **trainable:** True
8
- * **patch_size:** 14
9
- * **num_layers:** 12
10
- * **hidden_dim:** 768
11
- * **num_heads:** 12
12
- * **intermediate_dim:** 3072
13
- * **layer_scale_init_value:** 1.0
14
- * **num_register_tokens:** 0
15
- * **use_mask_token:** True
16
- * **use_swiglu_ffn:** False
17
- * **dropout_rate:** 0.0
18
- * **drop_path_rate:** 0.0
19
- * **image_shape:** [518, 518, 3]
20
- * **position_embedding_shape:** [518, 518]
21
- * **antialias_in_interpolation:** False
22
-
23
- This model card has been generated automatically and should be completed by the model author. See [Model Cards documentation](https://huggingface.co/docs/hub/model-cards) for more information.
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
  ---
2
  library_name: keras-hub
3
  ---
4
+ ### Model Overview
5
+ Vision Transformer (ViT) model trained using the DINOv2 method.
6
+
7
+ **Reference**
8
+
9
+ - [Learning Robust Visual Features without Supervision](https://arxiv.org/abs/2304.07193)
10
+ - [Vision Transformers Need Registers](https://arxiv.org/abs/2309.16588)
11
+
12
+ DINOV2 offers a powerful, generalist visual backbone learned entirely from
13
+ unlabeled images as described in DINOv2: Learning Robust Visual Features
14
+ without Supervision
15
+
16
+ ## Links
17
+
18
+ * [DINOv2 Quickstart Notebook] - coming soon
19
+ * [DINOv2 API Documentation] - coming soon
20
+ * [DINOv2 Beginner Guide] - coming soon
21
+ * [KerasHub Model Publishing Guide](https://keras.io/guides/keras_hub/upload/)
22
+
23
+
24
+ ## Installation
25
+
26
+ Keras and KerasHub can be installed with:
27
+
28
+ ```
29
+ pip install -U -q keras-hub
30
+ pip install -U -q keras
31
+ ```
32
+
33
+ Jax, TensorFlow, and Torch come preinstalled in Kaggle Notebooks. For instructions on installing them in another environment see the [Keras Getting Started](https://keras.io/getting_started/) page.
34
+
35
+ ## Presets
36
+
37
+ The following model checkpoints are provided by the Keras team. Weights have been ported from: https://huggingface.co. Full code examples for each are available below.
38
+
39
+ | Preset name | Parameters | Description |
40
+ |------------------------------------|------------|-----------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------|
41
+ | dinov2_small | 22.58M | Vision Transformer (small-sized model) trained using DINOv2. |
42
+ | dinov2_base | 87.63M | Vision Transformer (base-sized model) trained using DINOv2. |
43
+ | dinov2_large | 305.77M | Vision Transformer (large-sized model) trained using DINOv2. |
44
+ | dinov2_giant | 1.13B | Vision Transformer (giant-sized model) trained using DINOv2.|
45
+ | dinov2_with_registers_small | 22.58M | Vision Transformer (small-sized model) trained using DINOv2, with registers. |
46
+ | dinov2_with_registers_base | 87.63M | Vision Transformer (base-sized model) trained using DINOv2, with registers. |
47
+ | dinov2_with_registers_large | 305.77M | Vision Transformer (large-sized model) trained using DINOv2, with registers. |
48
+ | dinov2_with_registers_giant | 1.13B | Vision Transformer (giant-sized model) trained using DINOv2, with registers.|
49
+