setup env

Author: dimidagd

Makefile

@@ -135,3 +135,15 @@ build-release:
 	python setup.py bdist_wheel
 	python setup.py sdist
 	python utils/check_build.py
+
+
+get_checkpoints:
+	git clone https://github.com/facebookresearch/dinov3.git | echo "exists"
+	pip install -r  dinov3/requirements.txt &
+	wget -O /tmp/backbone.pth "https://dinov3.llamameta.net/dinov3_vit7b16/dinov3_vit7b16_pretrain_lvd1689m-a955f4ea.pth?Policy=eyJTdGF0ZW1lbnQiOlt7InVuaXF1ZV9oYXNoIjoiajEybnhqNWl2cGVtZDhvMDc0ZzF0bTFtIiwiUmVzb3VyY2UiOiJodHRwczpcL1wvZGlub3YzLmxsYW1hbWV0YS5uZXRcLyoiLCJDb25kaXRpb24iOnsiRGF0ZUxlc3NUaGFuIjp7IkFXUzpFcG9jaFRpbWUiOjE3NjMyMjM1NDd9fX1dfQ__&Signature=FkDkgcYMlUa1Oq%7E-emh1yXIIuj89OBh9O8PHcPhTEzw5MHLwggBnLQYGwbPGyLySCz8sfLN2YyXnk47gHHZLITGv%7EkFCGG4cFHN0inQIUqZIwRkghz9QMRqAAIxL1VnajHpGlfmewPhpS8Dawi8V99LOwZ3YQ9GKq3Uif5Re98VXgkL3Qj0KvMXvA%7Ez7w5zh8ZPCW3ggVDsKAf1P-Y66sohOQEwuCQbkycqwwXsXPkw%7EPSw68Ct9dpASlIpXZp-4SFobbTtvbpQ2C6R0E8M7OOkFVH4%7E%7E0W-n-xJpkSDdfrm7B2BWjZ2eDBMI3w4kcexVICP6smbFUZUbIY3uXIV8Q__&Key-Pair-Id=K15QRJLYKIFSLZ&Download-Request-ID=1511531990180009"
+	wget -O /tmp/lc.pth "https://dinov3.llamameta.net/dinov3_vit7b16/dinov3_vit7b16_imagenet1k_linear_head-90d8ed92.pth?Policy=eyJTdGF0ZW1lbnQiOlt7InVuaXF1ZV9oYXNoIjoiajEybnhqNWl2cGVtZDhvMDc0ZzF0bTFtIiwiUmVzb3VyY2UiOiJodHRwczpcL1wvZGlub3YzLmxsYW1hbWV0YS5uZXRcLyoiLCJDb25kaXRpb24iOnsiRGF0ZUxlc3NUaGFuIjp7IkFXUzpFcG9jaFRpbWUiOjE3NjMyMjM1NDd9fX1dfQ__&Signature=FkDkgcYMlUa1Oq%7E-emh1yXIIuj89OBh9O8PHcPhTEzw5MHLwggBnLQYGwbPGyLySCz8sfLN2YyXnk47gHHZLITGv%7EkFCGG4cFHN0inQIUqZIwRkghz9QMRqAAIxL1VnajHpGlfmewPhpS8Dawi8V99LOwZ3YQ9GKq3Uif5Re98VXgkL3Qj0KvMXvA%7Ez7w5zh8ZPCW3ggVDsKAf1P-Y66sohOQEwuCQbkycqwwXsXPkw%7EPSw68Ct9dpASlIpXZp-4SFobbTtvbpQ2C6R0E8M7OOkFVH4%7E%7E0W-n-xJpkSDdfrm7B2BWjZ2eDBMI3w4kcexVICP6smbFUZUbIY3uXIV8Q__&Key-Pair-Id=K15QRJLYKIFSLZ&Download-Request-ID=1511531990180009"
+
+# Load checkpoints and verify they are accessible
+
+load_checkpoints:
+	python test_load.py

test_load.py

@@ -0,0 +1,20 @@
+import torch
+import os
+
+checkpoint_paths = {
+    'backbone': '/tmp/backbone.pth',
+    'linear_head': '/tmp/lc.pth'
+}
+
+for name, path in checkpoint_paths.items():
+    if os.path.exists(path):
+        try:
+            checkpoint = torch.load(path, map_location='cpu')
+            print(f"✓ Successfully loaded {name} checkpoint from {path}")
+            print(f"  Checkpoint keys: {list(checkpoint.keys()) if isinstance(checkpoint, dict) else 'tensor'}")
+        except Exception as e:
+            print(f"✗ Failed to load {name} checkpoint: {e}")
+    else:
+        print(f"✗ {name} checkpoint not found at {path}")
+
+

test_torchhubload.py

@@ -0,0 +1,4 @@
+import torch
+
+# DINOv3
+dinov3_vit7b16_lc = torch.hub.load("./dinov3", 'dinov3_vit7b16_lc', source="local", weights="/tmp/lc.pth", backbone_weights="/tmp/backbone.pth")