kmjtspxr
/

minjin

minjin / transform.py

14:22

d1de587 almost 2 years ago

157 Bytes

from ultralytics import YOLO

# 모델 로드
model = YOLO('best.pt')

# ONNX로 직접 내보내기
model.export(format='onnx', simplify=True, dynamic=True)

	from ultralytics import YOLO

	# 모델 로드
	model = YOLO('best.pt')

	# ONNX로 직접 내보내기
	model.export(format='onnx', simplify=True, dynamic=True)