본문 바로가기
Programming/.NET

[.NET] ML.NET 고양이 vs 개 분류(Image Classification)

by 째스터 2023. 2. 5.
728x90

거의 모든 것을 할 수 있는 .NET 으로 머신러닝을 찍먹해본다.
아래는 10분 튜토리얼이다.
https://dotnet.microsoft.com/en-us/learn/ml-dotnet/get-started-tutorial/intro

 

ML.NET Tutorial | Get started in 10 minutes | .NET

Step-by-step instructions for building a simple prediction model with ML.NET on Windows, Linux, or macOS. ML.NET is a machine learning framework for .NET.

dotnet.microsoft.com

근데 설치만으로 10분 넘게 걸린듯 하다.


나는 튜토리얼과 다르게 개 vs 고양이를 구별하는 모델을 만들어볼 예정이다.

1. 먼저 C# 콘솔 애플리케이션 솔루션을 추가한다.
2. 아래와 같이 Machine Learning Model을 추가한다.

3. 고양이와 강아지가 분류된 데이터 경로를 입력한다.
(데이터셋은 여기서 다운로드: https://www.microsoft.com/en-us/download/details.aspx?id=54765)

4. 트레이닝 시작

평균 학습 시간은 다음과 같다. 나는 약 800MB를 학습할 예정이니 약 1시간을 기다려야 한다.

5. 학습 완료(41분정도 걸렸다.)

6. 후추 사진으로 테스트 완료. 너무 신기하고...

7. 해당 모델을 사용할 수 있도록 자동으로 프로젝트를 생성해준다.
(API와 Console App을 선택할 수 있다.)

아래 repository에서 확인할 수 있다.
https://github.com/HanJaeJoon/ML.NET-cats-vs-dogs-classification

 

GitHub - HanJaeJoon/ML.NET-cats-vs-dogs-classification: ML.NET Dogs vs Cats Classification Example

ML.NET Dogs vs Cats Classification Example. Contribute to HanJaeJoon/ML.NET-cats-vs-dogs-classification development by creating an account on GitHub.

github.com

 

728x90

댓글