[swift] Core ML이 Swift 프로젝트에 어떻게 통합되나요?

11 Dec 2023

swift

Core ML을 Swift 프로젝트에 통합하는 방법

1. Core ML 모델 파일 추가

Core ML 모델 파일을 Xcode 프로젝트에 추가합니다. .mlmodel 확장자를 가진 모델 파일을 Xcode 프로젝트 내에 드래그 앤 드롭하여 추가할 수 있습니다.

2. 모델 코드 생성

Xcode에서 모델 파일을 선택하고, 메뉴에서 “Editor” > “Create ML Model Class…“를 선택하여 Swift 클래스 코드를 생성합니다.

3. 모델 통합

생성된 모델 클래스를 사용하여 Swift 코드에서 모델을 로드하고 예측을 수행합니다.

import CoreML

// 모델 로드
guard let model = try? YourGeneratedModelClass(configuration: MLModelConfiguration.init()) else {
    fatalError("모델을 로드할 수 없음")
}

// 입력 데이터 생성
let inputData = YourModelInputDataType(...) // 모델에 필요한 입력 데이터 생성

// 예측 수행
if let prediction = try? model.prediction(input: inputData) {
    // 예측 결과 사용
    let outputData = prediction.output // 모델의 출력 데이터
    // 결과 처리
} else {
    fatalError("예측에 실패함")
}

Swift 프로젝트에 Core ML을 통합하는 것은 모델 파일의 추가, 모델 클래스 생성 및 모델을 사용하여 예측을 수행하는 단계로 간단히 완료됩니다.

더 자세한 내용 및 예제는 Apple의 Core ML 문서를 참조하십시오.