[Example] iOS AI 예제 #1 - Image Classifier (MLModel)

[Example] iOS AI 예제 #0

 

Image Classifier 화상분류 : 이미지가 속하는 Class(ex. dog, coffee)를 예측

 

1. ImageClassifier

교재 예제 Github 에서 제공하는 .mlmodel 파일을 사용했다. 

사진을 입력하면 class를 나타내는 label(String)을 제공하고 해당 label key를 넣으면 신뢰도 Value를 반환해주는 예측 모델이다.

 

2. SwiftUI ImagePicker 예제

아무래도 이 예제는 자주 사용하게 될 것 같다.

 

3. 모델 사용 예제

VisionKit API를 사용하는 것과 거의 유사하게 동작한다.

CompletionHandler 형태로도 할 수 있고 나는 SwiftUI로 코드를 짜고 싶어서 위 형태로 구현했다.

 

요즘 힐링 "보통견주" 보통이~

정말 간단하게 구현했다.

버튼을 누르면 ImagePicker가 뜨고 이미지를 선택하면

해당 이미지를 보여주면서 Classifier 추측을 하고

MLModel에서 반환해준 결과를 보여준다.

 

예제 코드 보기 Github

 

끗~

 

 

 

Reference by : iPhone/Android 애플리케이션 개발자를 위한 머신러닝, 딥러닝 실전입문 github

iPhone/Android 애플리케이션 개발자를 위한 머신러닝, 딥러닝 실전입문 - YES24

GitHub - TurningPointPub/ml-for-app-devs: iPhone/Android 애플리케이션 개발자를 위한 머신러닝, 딥러닝 실전