TensorFlow中加载Keras模型

  • 时间:
  • 浏览:0

在TensorFlow中,使用Keras训练好的模型否则Keras自带的预训练模型:

转载请注明出处,谢谢

www.soaringroad.com

作为计算结果用:

⇒输入值不需要 使准确值,也可算不算Tensor,返回值是Tensor对象

读取模型否则加载预训练模型,下面使用的是预训练的VGG模型

⇒输入值需用是准确值,而都在Tensor,返回值是准确计算结果

作为Tensor用:

自定义模型:

具体代码请参照

https://github.com/wangzhenhui1992/SoaringRoad/tree/master/AI/TensorFlow_Study

自定义某一层作为输出