今天在跑高光谱分类SOTA模型SpectralNET《SpectralNET a 2D wavelet CNN for Hyperspectral Image Classification》时遇到一个问题。
报错信息:
ImportError: cannot import name 'get_config' from 'tensorflow.python.eager.context'
报错原因:Keras版本不符
模型中的Keras版本为2.4.3,我安装的时候直接
pip install Keras默认安装的是最新版的2.8.0
解决方法:
方法一:
pip install keras==xxx(模型原始的版本)
或者
pip install git+https://www.github.com/keras-team/keras-contrib.git
方法二:
pip install --upgrade tensorflow pip install --upgrade tensorflow-gpu
我使用第一种方法,问题解决。
获取更多AI特辑:AiCharm