使用细分模型库时出现的问题

问题描述

from tensorflow import keras
from segmentation_models import PSPNet

运行此代码时，我面临以下错误：

AttributeError                            Traceback (most recent call last)
<ipython-input-14-fbd9360b4944> in <module>()
      1 
      2 from tensorflow import keras
----> 3 from segmentation_models import PSPNet

3 frames
/usr/local/lib/python3.6/dist-packages/segmentation_models/__init__.py in <module>()
     96 _framework = os.environ.get('SM_FRAMEWORK',_DEFAULT_KERAS_FRAMEWORK)
     97 try:
---> 98     set_framework(_framework)
     99 except ImportError:
    100     other = _TF_KERAS_FRAMEWORK_NAME if _framework == _KERAS_FRAMEWORK_NAME else _KERAS_FRAMEWORK_NAME

/usr/local/lib/python3.6/dist-packages/segmentation_models/__init__.py in set_framework(name)
     66     if name == _KERAS_FRAMEWORK_NAME:
     67         import keras
---> 68         import efficientnet.keras  # init custom objects
     69     elif name == _TF_KERAS_FRAMEWORK_NAME:
     70         from tensorflow import keras

/usr/local/lib/python3.6/dist-packages/efficientnet/keras.py in <module>()
     15 preprocess_input = inject_keras_modules(model.preprocess_input)
     16 
---> 17 init_keras_custom_objects()

/usr/local/lib/python3.6/dist-packages/efficientnet/__init__.py in init_keras_custom_objects()
     69     }
     70 
---> 71     keras.utils.generic_utils.get_custom_objects().update(custom_objects)
     72 
     73 

AttributeError: module 'keras.utils' has no attribute 'generic_utils'

我使用pip作为提供的说明（link）安装了细分模型库 如果有人可以帮助我解决该问题，我将不胜感激。我只是复制指令中的代码，所以我在网上发现的一切都是一样的。安装是否应该有问题？

请引导我完成此操作！：）

解决方法

由于使用Tensorflow 2.1/2.0，因此您面临此问题。要解决此问题，您必须使用Tensorflow 1.x (i.e 1.15.2)或Image segmentation

请按照以下步骤使用Segmentation models和TF 2.1使用 !pip install q tensorflow==2.1 !pip install segmentation-models import tensorflow as tf from segmentation_models import PSPNet #Instantiate PSPNet model model = PSPNet() #Display model summary model.summary() 。

overflow: hidden