[Computer Vision Examples] Convert PyTorch: pt to onnx

Convert PyTorch: pt to onnx

포스트 난이도: HOO_Senior

---

# pt to onnx

아래는 onnx로 전환하는 예제코드이다.

Colab이나 Notebook에서 작업할 수 있고 vscode에서도 notebook이 있으면 가능하다.

사용 조건에 맞춰서 조정해서 사용하면 된다.

---

!pip install ultralytics onnx onnx-simplifier onnxruntime

---

from ultralytics import YOLO
model = YOLO("model_name.pt") 
model.export(format="onnx",imgsz=[640,640], opset=12)  # export the model to ONNX format

 

---