Upload README.md with huggingface_hub
Browse files
README.md
ADDED
|
@@ -0,0 +1,32 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
---
|
| 2 |
+
license: mit
|
| 3 |
+
tags:
|
| 4 |
+
- onnx
|
| 5 |
+
- perceptual-hashing
|
| 6 |
+
- dinohash
|
| 7 |
+
---
|
| 8 |
+
|
| 9 |
+
# DINOHash — extra checkpoints
|
| 10 |
+
|
| 11 |
+
Additional DINOHash perceptual-hashing models not present in
|
| 12 |
+
[`backslashh/DINOHash`](https://huggingface.co/backslashh/DINOHash).
|
| 13 |
+
Each model is provided both as the **raw** training/traced artifact (`raw/`)
|
| 14 |
+
and as an exported **ONNX** graph (repo root, dynamic batch axis, opset 17).
|
| 15 |
+
|
| 16 |
+
| Model | ONNX | Raw | Notes |
|
| 17 |
+
|---|---|---|---|
|
| 18 |
+
| ViT-Small → ViT-Tiny (DINO distill) | `ViT-Small-ViT-Tiny.onnx` | `raw/ViT-Small-ViT-Tiny.pth` | student backbone (`vit_tiny_patch16_224`), 192-d embedding |
|
| 19 |
+
| XCiT-Small → XCiT-Tiny (DINO distill) | `XCiT-Small-XCiT-Tiny.onnx` | `raw/XCiT-Small-XCiT-Tiny.pth` | student backbone (`xcit_tiny_12_p16_224`), 192-d embedding |
|
| 20 |
+
| MAE-Lite mae_tiny_400e | `mae_tiny_400e_traced.onnx` | `raw/mae_tiny_400e_traced.pt` | 192-d |
|
| 21 |
+
| MAE-Lite mae_tiny_distill_400e | `mae_tiny_distill_400e_traced.onnx` | `raw/mae_tiny_distill_400e_traced.pt` | 192-d |
|
| 22 |
+
| MAE-Lite mae_tiny_distill_d2_400e | `mae_tiny_distill_d2_400e_traced.onnx` | `raw/mae_tiny_distill_d2_400e_traced.pt` | 192-d |
|
| 23 |
+
| MAE-Lite mocov3_tiny_400e | `mocov3_tiny_400e_traced.onnx` | `raw/mocov3_tiny_400e_traced.pt` | 192-d |
|
| 24 |
+
|
| 25 |
+
## Notes on the raw files
|
| 26 |
+
- **MAE-Lite** raw files are TorchScript (`_traced.pt`), self-contained and loadable directly.
|
| 27 |
+
- **ViT / XCiT** raw files are full DINO training checkpoints (`student`/`teacher`/`optimizer`/...).
|
| 28 |
+
The ONNX graphs were built by extracting the `student.backbone.*` weights into the matching
|
| 29 |
+
`timm` architecture (strict-clean load) and exporting; XCiT required `pos_embeder`→`pos_embed`
|
| 30 |
+
rename and qkv split/fuse between class-attention and XCA blocks.
|
| 31 |
+
|
| 32 |
+
All inputs are `(batch, 3, 224, 224)`.
|