ImportError: cannot import name '_obtain_input_shape' in keras

ImportError: cannot import name '_obtain_input_shape' in keras

当我尝试导入 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 1, in <module>
    from keras.applications.imagenet_utils import _obtain_input_shape
ImportError: cannot import name '_obtain_input_shape'

导入代码如下:

import tensorflow as tf
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

如何解决? (我使用的是tensorflow==2.6.2,keras==2.6.0,keras-squeezenet==0.4)。

你试过新版本了吗? (参见:https://github.com/rcmalli/keras-squeezenet

你可以安装它:
pip 安装 git+https://github.com/rcmalli/keras-squeezenet.git