java - DL4J:设置 "channels_first"配置

Question

我正在尝试使用 DL4J 将预训练的 Keras 模型加载到 Java 应用程序中。在 Keras 中训练时，我使用了“channels_first”image_data_format，现在，在 DL4J 中我无法正确加载模型，因为它尝试使用“channels_last”配置。

DL4J 中是否有与 Keras 中的 set_image_data_format("channels_first) 类似的函数？

Answer 1

您无法设置这样的属性，因为您根本不需要这样做。 DL4J 的 Keras 模型导入的设计方式是，您提供 json 或 h5 文件，DL4J 将确定您正在使用哪个 Keras 版本、哪个后端，特别是哪种数据格式。

请参阅此处了解 API 简介: https://deeplearning4j.org/model-import-keras

以下是支持的功能列表: https://deeplearning4j.org/keras-supported-features

为了帮助您入门，您可能需要查看一些测试，例如 model configuration或end to end tests .

如果您有任何疑问，请随时通过 gitter 联系我。另外，请随意创建 issue如果你有的话。