Upload convert_safetensor_to_pt.py

convert_safetensor_to_pt.py

@@ -0,0 +1,9 @@
+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')