我一直在尝试使用 coreML 创建一个 IOS 应用程序,我在 Keras 中训练了一个卷积神经网络,当我使用 CoreMLTools 将此模型转换为 CoreML 模型时,它显示输出是一个多维数组,但我希望它是类概率。如何在 Keras 中生成带有类标签的 .txt 文件?
这是我用来生成 coreML 模型的代码:
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')