我正在尝试使用DL4J将预训练的Keras模型加载到 Java 应用程序中。在 Keras 中训练时，我使用了“channels_first”image_data_format，现在，在 DL4J 中我无法正确加载我的模型，因为它尝试使用“ channels_last ”配置。
DL4J 中是否有与 Keras 中的set_image_data_format("channels_first)类似的功能？
问问题
67 次
1 回答
1
您不能设置这样的属性，因为您根本不需要。DL4J 的 Keras 模型导入的设计方式是您提供您的jsonorh5文件，DL4J 将确定您使用的是哪个 Keras 版本、哪个后端，尤其是哪种数据格式。
有关 API 的介绍，请参见此处： https ://deeplearning4j.org/model-import-keras
以下是支持的功能列表： https ://deeplearning4j.org/keras-supported-features
为了让您开始，您可能需要查看一些测试，例如这些模型配置或端到端测试。
如果您有任何问题，请随时在 gitter 上联系我。此外，如果您有问题，请随时创建问题。
于 2018-04-19T14:00:54.373 回答