gajeshladhar
/

core-dino

Model card Files Files and versions

gajeshladhar commited on Jun 27

Commit

d2997bb

·

1 Parent(s): c5e40a5

readme.md updated

Files changed (1) hide show

README.md +3 -2

README.md CHANGED Viewed

@@ -27,10 +27,11 @@ license: cc-by-nc-4.0
 ```python
 from ultralytics import YOLO
-ckpt = "https://huggingface.co/gajeshladhar/core-dino/resolve/main/checkpoints/student.pt"
 model = YOLO("yolo11x-obb.pt") # obb, bbox or seg any model
 model.model.load_state_dict(
-    {k.replace('layers.', 'model.'): v for k, v in torch.hub.load_state_dict_from_url(ckpt, map_location='cpu').items()},
     strict=False)
 ```

 ```python
 from ultralytics import YOLO
 model = YOLO("yolo11x-obb.pt") # obb, bbox or seg any model
+ckpt = "https://huggingface.co/gajeshladhar/core-dino/resolve/main/checkpoints/student.pt"
+ckpt = torch.hub.load_state_dict_from_url(ckpt, map_location='cpu')
 model.model.load_state_dict(
+    {k.replace('layers.', 'model.'): v for k, v in ckpt.items()},
     strict=False)
 ```