YOLOv5 모델을 직접 ONNX로 변환 (윈도우에서 변환)

 

사전 변환된 파일이 신뢰되지 않거나, 최신 모델을 사용하려면 직접 변환할 수도 있습니다.

📌 설치 및 환경 설정

1. Python 3.8 이상 설치 (다운로드 링크)
2. YOLOv5 저장소 다운로드
   bash

   git clone https://github.com/ultralytics/yolov5 cd yolov5
3. 필수 패키지 설치
   bash

   복사편집

   pip install -r requirements.txt

📌 YOLOv5s ONNX 변환

1. PyTorch 모델을 ONNX로 변환
   bash

   python export.py --weights yolov5s.pt --include onnx
2. 변환이 완료되면 yolov5s.onnx 파일이 yolov5/runs/ 폴더에 생성됩니다.

---

📌 변환된 ONNX 파일 위치 및 활용

• yolov5s.onnx 파일을 C# 프로젝트에서 사용하려면:
  • 변환 후 C:\Models\yolov5s.onnx 폴더에 복사
  • C#에서 ONNX Runtime을 이용해 실행

C# 코드 예제 (YOLOv5 ONNX 실행):

using System;
using System.IO;
using System.Linq;
using Microsoft.ML.OnnxRuntime;
using Microsoft.ML.OnnxRuntime.Tensors;

class Program
{
    static void Main()
    {
        string modelPath = @"C:\Models\yolov5s.onnx";
        using var session = new InferenceSession(modelPath);

        Console.WriteLine("YOLOv5s ONNX 모델 로드 성공!");
    }
}

---