isaaccorley/core-dino

Rehosted and modified from https://huggingface.co/gajeshladhar/core-dino

Modified using the following code:

import os
import hashlib

import torch
import ultralytics

model = ultralytics.YOLO("yolo11x.yaml")
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)

model.ckpt = model.state_dict()

filename = "core_dino_rgb_yolo11x.pt"
model.save(filename)
md5 = hashlib.md5(open(filename, "rb").read()).hexdigest()[:8]
os.rename(filename, filename.replace(".pt", f"-{md5}.pt"))