我正在尝试使用DL4J将经过预先训练的Keras模型加载到Java应用程序中。在Keras中进行培训时,我使用了"channels_first“image_data_format,而现在在DL4J中,我无法正确加载我的模型,因为它试图使用channels_first配置。
在DL4J中有类似于Keras中的set_image_data_format("channels_first)的函数吗?
发布于 2018-04-19 14:00:54
您不能设置这样的属性,因为您不需要这样做。用于DL4J的Keras模型导入的设计方式是提供json或h5文件,DL4J将确定您使用的Keras版本、哪个后端,特别是哪种数据格式。
有关API:https://deeplearning4j.org/model-import-keras的介绍,请参见此处
下面是支持的特性列表:https://deeplearning4j.org/keras-supported-features
为了让您开始工作,您可能需要查看一些测试,比如这些模型配置或端到端测试。
如果你有什么问题的话,可以随便给我打电话。此外,如果您有问题,可以自由地创建一个。
https://stackoverflow.com/questions/49492046
复制相似问题