转移学习模型不适用于android(训练后)

时间:2018-04-09 20:09:02

标签: android machine-learning keras

我使用转移学习来学习一系列功能。所以在我的转学习任务中使用了这3个最后一层。

图层(类型)输出形状参数#

dense_1(密集)(无,256)6422784

dropout_1(Dropout)(None,256)0

dense_2(密集)(无,2)514

总参数:6,423,298 可训练的参数:6,423,298 不可训练的参数:0

在ipythonNotebook上进行所有锻炼但是在将其制作成.pb文件并加载到Android Studio之后。发生此错误

  

不是有效的TensorFlow图形序列化:NodeDef预期输入''不匹配指定的1个输入;操作输出:dtype; ATTR =值:张量; ATTR = D型:类型取代; NodeDef:dropout_1 / cond / dropout / random_uniform / max = Constdtype = DT_FLOAT,value = Tensor

相同的代码适用于在第一层输入的模型 _例如:conv2d_input_1_这是我以前的任务,与转学不相关,但一旦模型改变了代码就行不通。我认为tensorflowAndroid希望输入层能够提供图像,但是我们的传输学习模型没有输入层来提供输入并且每次都显示问题 我检查了android上的tensorflow和tensorflow的版本并匹配了版本,但仍然没有运气。你能帮帮我!!

0 个答案:

没有答案