10 lines
319 B
Python
10 lines
319 B
Python
import torch
|
|
from ultralytics.nn.tasks import DetectionModel
|
|
from safetensors.torch import load_file
|
|
|
|
tensor_dict = load_file("weights/icon_detect/model.safetensors")
|
|
|
|
model = DetectionModel('weights/icon_detect/model.yaml')
|
|
model.load_state_dict(tensor_dict)
|
|
torch.save({'model':model}, 'weights/icon_detect/best.pt')
|