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
使用以下命令以使用预训练的挤压网络模型：
第一：
!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
注意：链接可能会过期。

从keras_squeezenet导入squeezenet导入时，出现导入错误

1 个答案: