python - 从 keras_squeezenet 导入时，我收到了导入错误

Question

导入时出现此错误

ImportError                               Traceback (most recent call last)
<ipython-input-30-5f3b3ff11ebd> in <module>
      5 import os
      6 from tensorflow.keras.utils import to_categorical
----> 7 from keras_squeezenet import squeezenet

~\Anaconda3\envs\python3.7-TF2.0\lib\site-packages\keras_squeezenet\__init__.py in <module>
----> 1 from keras_squeezenet.squeezenet import SqueezeNet
      2 from keras_squeezenet.version import __version__

~\Anaconda3\envs\python3.7-TF2.0\lib\site-packages\keras_squeezenet\squeezenet.py in <module>
----> 1 from keras.applications.imagenet_utils import _obtain_input_shape
      2 from keras import backend as K
      3 from keras.layers import Input, Convolution2D, MaxPooling2D, Activation, concatenate, Dropout, warnings
      4 from keras.layers import GlobalAveragePooling2D, GlobalMaxPooling2D
      5 from keras.models import Model

ImportError: cannot import name '_obtain_input_shape' from 'keras.applications.imagenet_utils'

我正在使用 TensorFlow 2.0.0 版和 keras 2.2.4-tf 版

我可以通过 tf.keras 或任何其他类似的方式导入 SqueezeNet 模型吗？

Answer 1

使用 tensorflow 版本 1.14.0rc1
和 keras 版本 2.1.1
使用以下命令来使用预训练的squeezenet模型：
首先：
!pip install git+https://github.com/rcmalli/keras-squeezenet.git
第二：
!pip install tensorflow==1.14.0rc1
然后：
!pip install q keras==2.1.1
最后你可以使用：

from keras_squeezenet import SqueezeNet

检查我建立的模型：
https
://colab.research.google.com/drive/1nias2agRtn0G0Wn0KouJhkyTMO7IQkIS?usp=sharing 注意：链接可能会过期。