使用TensorFlow.js转换器转换的模型无法运行求助
问题:TFJS加载转换后的MobileNet模型时提示InputLayer需传入batchInputShape或inputShape
我正在学习Jason Mayes的TFJS WebML YouTube课程,跟着其中的TFJS Converter视频在Google Colab中操作,步骤和视频基本一致,仅出现少量额外警告,成功生成了MobileNetV2模型文件。
但在网页中使用TFJS的tf.loadLayersModel('MODEL.JSON的URL')加载模型时,收到错误提示:
InputLayer应传入batchInputShape或inputShape
尽管model.json文件中存在batch_shape字段,且我已经将TFJS更新到最新版本,还是找不到问题根源。
之后我尝试转换MobileNetV3-Large,将Colab笔记本中的tf.keras.applications.MobileNetV2替换为tf.keras.applications.MobileNetV3Large,生成模型后遇到了完全相同的错误。
我目前正在学习TensorFlow/TensorFlow.js和机器学习相关知识,希望能得到解决这个问题的帮助,谢谢!
内容的提问来源于stack exchange,提问作者Dhanwanth Parameswar
相关产品推荐
相关产品推荐

