DL4J：设置“channels_first”config

Question

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

在DL4J中是否有与Keras中 set_image_data_format（“channels_first）类似的功能？

Answer 1

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

有关API的介绍，请参阅此处： https://deeplearning4j.org/model-import-keras

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

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

如果您有任何疑问，请随时给我打电话。另外，如果您有issue，请随时创建{{3}}。