为了账号安全,请及时绑定邮箱和手机立即绑定

导入错误:无法导入名称“relu6”和 AttributeError:

导入错误:无法导入名称“relu6”和 AttributeError:

慕森卡 2021-07-02 10:03:42
我正在尝试.mlmodel使用以下代码将 h5 keras 模型转换为文件类型:from keras.models import load_modelimport kerasfrom keras.applications.mobilenet import MobileNetfrom keras.layers import DepthwiseConv2D# convert the model to coreml formatprint("[INFO] converting model")from keras.utils.generic_utils import CustomObjectScopewith CustomObjectScope({'relu6': keras.applications.mobilenet.relu6,'DepthwiseConv2D':     keras.applications.mobilenet.DepthwiseConv2D}):   model = load_model('/Users/nikhil.c/aslModel.h5', custom_objects={               'relu6': MobileNet})coreml_model = coremltools.converters.keras.convert("/Users/nikhil.c        /aslModel.h5",        input_names="image",        image_input_names="image",        image_scale=1/255.0,        class_labels= ["hello", "hi", "you"],        is_bgr=True)# save the model to diskcoremltools.utils.save_spec(coreml_model, 'aslModel.mlmodel')在使用之前,我最初收到此错误CustomObjectScope:ImportError: cannot import name 'relu6'我通过修复它CustomObjectScope,但现在我收到错误:AttributeError: module 'keras.applications.mobilenet' has no attribute 'relu6'. 我通常不会在堆栈溢出中发帖,所以如果您需要更多信息,请告诉我。
查看完整描述

1 回答

?
Cats萌萌

TA贡献1805条经验 获得超9个赞

您拥有的代码适用于较旧的 Keras 版本,我检查的最新 Keras (2.2.2) 已经集成了 ReLU 和 DepthWiseConv2D keras.layers,因此您只需要导入它即可使用 MobileNet:


import keras

from keras.applications import MobileNet

MobileNetV2更新版本的 MobileNet也在同一个包中。


查看完整回答
反对 回复 2021-07-06
  • 1 回答
  • 0 关注
  • 398 浏览
慕课专栏
更多

添加回答

举报

0/150
提交
取消
意见反馈 帮助中心 APP下载
官方微信