如何解决tf.keras 模型到 coreml 模型,不支持 BatchNormalization
我正在从事一个图像识别项目,该项目使用来自 keras(tensorflow 版本)的 InceptionV3。我已经训练了一个输出层,现在我正在尝试将其导出为 coreMl 格式,以便在 ios 设备上使用它。
然而,我得到了错误:
ValueError:不支持 Keras 层“
当我跑步时:
model.save('IncepNet_SN_class.h5')
output_labels = ['0','1','2','3','4']
model_2 = tf.keras.models.load_model('IncepNet_SN_class.h5')
your_model = coremltools.converters.keras.convert('IncepNet_SN_class.h5',input_names=['image'],output_names=['output'],class_labels=output_labels)
your_model.save('IncepNet_SN_class.mlmodel')
有人知道怎么处理吗?
非常感谢,卢卡斯。
版权声明:本文内容由互联网用户自发贡献,该文观点与技术仅代表作者本人。本站仅提供信息存储空间服务,不拥有所有权,不承担相关法律责任。如发现本站有涉嫌侵权/违法违规的内容, 请发送邮件至 dio@foxmail.com 举报,一经查实,本站将立刻删除。