【问题标题】:While importing from keras_squeezenet import squeezenet I am getting import error从 keras_squeezenet 导入时,我收到了导入错误
【发布时间】:2020-11-28 18:22:28
【问题描述】:

导入时出现此错误

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 模型吗?

【问题讨论】:

  • 这是一个 Keras 版本问题。您可能需要降级 Keras 或寻找替代解决方案。

标签: python keras tensorflow2.0


【解决方案1】:

使用 tensorflow 版本 1.14.0rc1
和 keras 版本 2.1.1
使用以下命令来使用预训练的挤压网模型:
first:
!pip install git+https://github.com/rcmalli/keras-squeezenet.git
second:
!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
注意:链接可能会过期。

【讨论】:

    猜你喜欢
    • 2022-08-08
    • 2020-11-09
    • 2019-08-15
    • 1970-01-01
    • 2021-05-23
    • 2021-10-11
    • 1970-01-01
    • 1970-01-01
    • 2018-09-12
    相关资源
    最近更新 更多