minjin / transform.py
mjxxkxx's picture
14:22
d1de587
raw
history blame contribute delete
157 Bytes
from ultralytics import YOLO
# 모델 로드
model = YOLO('best.pt')
# ONNX로 직접 내보내기
model.export(format='onnx', simplify=True, dynamic=True)