ImportError: cannot import name 'get_config' in keras_squeezenet

Question

我正在尝试将 keras_squeezenet 导入到我的项目中，但出现此错误：

Traceback (most recent call last):
  File "C:/Users/belog/drone_sees/train_model.py", line 3, in <module>
    from keras_squeezenet import SqueezeNet
  File "C:\Users\belog\AppData\Local\Programs\Python\Python36\lib\site-packages\keras_squeezenet\__init__.py", line 1, in <module>
    from keras_squeezenet.squeezenet import SqueezeNet
  File "C:\Users\belog\AppData\Local\Programs\Python\Python36\lib\site-packages\keras_squeezenet\squeezenet.py", line 2, in <module>
    from keras import backend as K
  File "C:\Users\belog\AppData\Local\Programs\Python\Python36\lib\site-packages\keras\__init__.py", line 25, in <module>
    from keras import models
  File "C:\Users\belog\AppData\Local\Programs\Python\Python36\lib\site-packages\keras\models.py", line 19, in <module>
    from keras import backend
  File "C:\Users\belog\AppData\Local\Programs\Python\Python36\lib\site-packages\keras\backend.py", line 36, in <module>
    from tensorflow.python.eager.context import get_config
ImportError: cannot import name 'get_config'

导入代码如下：

import cv2
import numpy as np
from keras_squeezenet import SqueezeNet
from keras.optimizers import Adam
from keras.utils import np_utils
from keras.layers import Activation, Dropout, Convolution2D, GlobalAveragePooling2D
from keras.models import Sequential
import tensorflow as tf

如何解决这个错误？ （我正在使用 keras-squeezenet==0.4）。

Answer 1

尝试升级 tensorflow：

pip install --upgrade tensorflow
pip install --upgrade tensorflow-gpu