Windows ML을 통해 Custom Vision에서 ONNX 모델 사용(미리 보기)

  • 아티클

Windows ML을 사용하여 Custom Vision Service에서 내보낸 ONNX 모델을 사용하는 방법을 알아봅니다(미리 보기). 예제 UWP 애플리케이션을 학습된 이미지 분류자와 함께 사용합니다.

필수 조건

  • Windows 10 버전 1809 이상
  • 빌드 17763 이상용 Windows SDK
  • 유니버설 Windows 플랫폼 개발 워크로드가 사용하도록 설정된 Visual Studio 2017 버전 15.7 이상.
  • PC에서 개발자 모드를 사용하도록 설정합니다. 자세한 내용은 개발에 디바이스 사용을 참조하세요.

예제 앱 정보

포함된 애플리케이션은 일반 Windows UWP 앱입니다. 컴퓨터에서 이미지를 선택하고 로컬로 저장된 분류 모델을 사용하여 처리할 수 있습니다. 모델에서 반환되는 태그 및 점수는 이미지 옆에 표시됩니다.

애플리케이션 가져오기

예제 애플리케이션은 GitHub의 Azure AI 서비스 ONNX Custom Vision 샘플 리포지토리에서 사용할 수 있습니다. 로컬 컴퓨터에 복제하고 Visual Studio에서 SampleOnnxEvaluationApp.sln을 엽니다.

애플리케이션 테스트

  1. F5 키를 사용하여 Visual Studio에서 애플리케이션을 시작합니다. 개발자 모드를 사용하도록 설정하라는 메시지가 표시될 수 있습니다.
  2. 애플리케이션이 시작되면 단추를 사용하여 채점할 이미지를 선택합니다. 기본 ONNX 모델은 다양한 유형의 플랑크톤을 분류하도록 학습되었습니다.

사용자 고유의 모델 사용

자체 이미지 분류자 모델을 사용하려면 다음 단계를 수행합니다.

  1. Custom Vision Service를 사용하여 분류자를 만들고 학습합니다. 이 작업을 수행하는 방법에 대한 지침은 분류자 만들기 및 학습을 참조하세요. 일반(컴팩트)과 같은 컴팩트 도메인 중 하나를 사용합니다.
    • 다른 도메인을 사용하는 기존 분류자가 있는 경우 프로젝트 설정에서 컴팩트로 변환할 수 있습니다. 그런 다음, 계속하기 전에 프로젝트를 다시 학습합니다.
  2. 모델을 내보냅니다. 성능 탭으로 전환하고 컴팩트 도메인으로 학습된 반복을 선택합니다. 표시되는 내보내기 단추를 선택합니다. 그런 다음, ONNX, 내보내기를 차례로 선택합니다. 파일이 준비되면 다운로드 단추를 선택합니다. 내보내기 옵션에 대한 자세한 내용은 모델 내보내기를 참조하세요.
  3. 다운로드한 .zip 파일을 열고 여기에서 model.onnx 파일의 압축을 풉니다. 이 파일에는 분류자 모델이 포함되어 있습니다.
  4. Visual Studio의 솔루션 탐색기에서 자산 폴더를 마우스 오른쪽 단추로 클릭하고 기존 항목 추가를 선택합니다. ONNX 파일을 선택합니다.
  5. 솔루션 탐색기에서 ONNX 파일을 마우스 오른쪽 단추로 클릭하고 속성을 선택합니다. 파일의 다음 속성을 변경합니다.
    • 빌드 작업 ->내용
    • 출력 디렉터리로 복사 ->변경된 내용만 복사
  6. 그런 다음, MainPage.xaml.cs를 열고 _ourOnnxFileName의 값을 ONNX 파일의 이름으로 변경합니다.
  7. F5를 사용하여 프로젝트를 빌드하고 실행합니다.
  8. 단추를 선택하여 평가할 이미지를 선택합니다.

다음 단계

Custom Vision 모델을 내보내고 사용하는 다른 방법을 알아보려면 다음 문서를 참조하세요.

Windows ML에서 ONNX 모델을 사용하는 방법에 대한 자세한 내용은 Windows ML을 통해 앱에 모델 통합을 참조하세요.