我们正在尝试将 .h5 Keras 模型转换为 .mlmodel 模型,我的代码如下:
from keras.models import load_model
import keras
from keras.applications import MobileNet
from keras.layers import DepthwiseConv2D
from keras.utils.generic_utils import CustomObjectScope
with CustomObjectScope({'relu6': keras.applications.mobilenet.relu6,'DepthwiseConv2D': keras.applications.mobilenet.DepthwiseConv2D}):
model = load_model('CNN_tourist_11.h5', custom_objects={'relu6': MobileNet})
output_labels = ['0', '1', '2', '3', '4', '5', '6', '7', '8', '9']
coreml_model = coremltools.converters.keras.convert("CNN_tourist_11.h5",
input_names="image",
image_input_names="image",
class_labels= output_labels,)
coremltools.utils.save_spec(coreml_model, 'place10.mlmodel')
我们查找了 6 天前提出的类似问题,我们也导入了 MobileNet,但仍然显示此错误:
AttributeError: module 'keras.applications.mobilenet' has no attribute 'relu6'
我的 Tensorflow 版本是 1.10.0,Keras 版本是 2.2.2
如果有人能就为什么它一直显示这个错误给我们建议,我们将非常感激,非常感谢。