[CoreML] Failed to load Vision ML model: Error Domain=com.apple.vis Code=15 "The model does not have a valid input feature of type image"

 

CoreML을 적용한 iOS 앱을 만들며 마주한 에러와 해결 방법을 정리해보려 한다.

파이썬 프로젝트에서 keras 모델을 생성하여 Xcode에 모델을 추가하기 위해 변환하는 과정을 진행하고 있었는데

coremltools 패키지를 이용하면 CoreML에서 쓰이는 형식인 .mlmodel로 변환이 가능하다.

 

 

 

 

추출된 mlmodel을 Xcode 프로젝트에 성공적으로 추가해주었다.

 

 

그리고 실행하여 이미지를 앨범에서 선택하여 분류를 시작하였는데 다음과 같은 에러가 떴다.

 

 

왜인지 검색해봤더니 모델의 입력 형식과 관련해서 설정이 되지 않아 나오는 에러인 듯 했다. 아무래도 모델을 추출하는 과정에서 설정이 잘못된 듯 하여 파이썬 코드를 다시 확인해보았다.

 

 

CoreML 모델로 변환 시 입력 텐서의 이름을 모델의 실제 입력 이름과 일치하게끔 수정했다.

 

에러가 해결된 모습..