yushg commited on
Commit
d975b60
·
verified ·
1 Parent(s): 0459acb

Upload README.md with huggingface_hub

Browse files
Files changed (1) hide show
  1. README.md +32 -0
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)`.