CoreML 모델 용으로 Keras에서 class_labels.txt를 어떻게 생성합니까?

Question

나는 coreML을 사용하여 IOS 응용 프로그램을 만들려고 노력했다. Keras에서 길쌈 신경망을 훈련 시켰는데 CoreMLTools를 사용하여이 모델을 CoreML 모델로 변환하면 출력이 다차원 배열임을 보여 주지만 원하는 것은 계급 확률. Keras에 클래스 레이블이있는 .txt 파일을 어떻게 생성합니까? 훈련 집합의 클래스의 순서로,

이

import coremltools 

coreml_model = coremltools.converters.keras.convert(
                "chars74kV3.0.h5", class_labels = "class_labels.txt", image_input_names= ['input'], input_names=['input'], image_scale=255.) 
coreml_model.author = 'Thijs van der Heijden' 
coreml_model.license = 'MIT' 
coreml_model.description = 'A basic Deep Convolutional Neural Network to classify handwritten letters.' 
coreml_model.input_description['input'] = 'A 64x64 pixel Image' 
coreml_model.save('chars74k.mlmodel') 

Answer 1

class_labels.txt 파일 그냥 한 줄에 하나의 레이블이있는 일반 텍스트 파일이어야합니다 :

이

내가 coreML 모델을 생성하기 위해 사용하는 코드입니다 . 예를 들어,

dog 
cat 
person 

클래스 0 "개"이었다 세 가지 수준의 네트워크 레이블 파일이 될 것입니다, 클래스 1 "고양이", 그리고 클래스 2 "사람"이었다. 이것이 공용 분류 데이터 세트 인 경우 데이터 세트에 해당 정보가 있어야하며 자신의 데이터 세트 인 경우 해당 매핑 파일을 만들어야합니다. 어쨌든 클래스 번호를 값과 연결하려면이 작업을 수행해야합니다.