convert_segm2det.py 248 Bytes
Newer Older
chenzk's avatar
v1.0  
chenzk committed
1
2
3
4
5
6
7
8
9
10
11
12
from ultralytics import YOLOE
import torch

det_model = YOLOE("yoloe-v8l.yaml")

state = torch.load("yoloe-v8l-seg.pt")

det_model.load(state["model"])
det_model.save("yoloe-v8l-seg-det.pt")

model = YOLOE("yoloe-v8l-seg-det.pt")
print(model.args)