3

我正在使用一个使用 128x128 像素输入图像的 Keras 网络,该网络在 chars74K 数据集上的准确率超过 85%。当我将此网络转换为 CoreML 模型时,结果总是 100% 确定但总是错误的,从来没有正确的字母。我的 Keras 网络的代码可以在这里找到:https ://github.com/thijsheijden/chars74kCNN

我用来转换为 CoreMLModel 的代码如下:

import coremltools
import h5py
import pandas

coreml_model = coremltools.converters.keras.convert(
                                                    "chars74kV4.0.h5", class_labels = "class_labels.txt", image_input_names= ['input'], input_names=['input'])
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 128x128 pixel Image'
coreml_model.save('chars74k.mlmodel')

我的 IOS 应用程序的代码可以在这里找到:https ://github.com/thijsheijden/Visionary

我将不胜感激任何帮助,因为我真的坚持这一点!提前致谢!

4

0 回答 0