keras normalize_data_format 错误

最新推荐文章于 2022-02-28 18:34:52 发布

原创最新推荐文章于 2022-02-28 18:34:52 发布 · 2.2k 阅读

0 ·

CC 4.0 BY-SA版权

文章标签：

#keras #normalize_data_format #error

keras 专栏收录该内容

1 篇文章

订阅专栏

本文针对Keras不同版本中normalize_data_format函数位置变化的问题，提供了一种判断并调用该函数的解决方案，确保代码在不同版本间稳定运行。

keras 由于版本更迭的原因导致normalize_data_format函数处于不同位置，可以通过下面的判断直接解决

if keras.__version__ > "2.2.0":
    from keras.backend import normalize_data_format
    data_format = normalize_data_format(data_format)
else:
    from keras.utils.conv_utils import normalize_data_format
    data_format = normalize_data_format(data_format)