时间戳

首页 > 解决方案 > AttributeError：模块'keras.utils'没有使用segmentation_models的属性'get_file'

python - AttributeError：模块'keras.utils'没有使用segmentation_models的属性'get_file'

问题描述

我正在尝试使用分段模型，但无法修复此错误。我已经搜索了这个特定的，但找不到答案。我正在使用 pycharm，这个错误链接到这个特定的代码行 BACKBONE = 'resnet34'

model1 = sm.Unet(BACKBONE, weights=None,
                 encoder_weights='imagenet',
                 classes=num_classes,
                 activation='softmax',
                 decoder_block_type = 'upsampling')

这也是第 83 个。我在文档中进行了搜索，显然 tensorflow keras 等的版本满足要求。考虑到我真的尝试以多种组合方式安装和卸载所有内容，以便让这段代码工作。谢谢大家的帮助和时间！下面是完整的错误，希望它可以帮助你！

    `Traceback (most recent call last):
      File "C:\Users\Giulia\PycharmProjects\multiclass_new\main.py", line 83, in <module>
        model1 = sm.Unet('resnet34', weights=None,
      File "C:\Users\Giulia\PycharmProjects\multiclass_new\venv\lib\site- 
   packages\segmentation_models\__init__.py", line 34, in wrapper
        return func(*args, **kwargs)
      File "C:\Users\Giulia\PycharmProjects\multiclass_new\venv\lib\site- 
   packages\segmentation_models\models\unet.py", line 221, in Unet
        backbone = Backbones.get_backbone(
      File "C:\Users\Giulia\PycharmProjects\multiclass_new\venv\lib\site- 
   packages\segmentation_models\backbones\backbones_factory.py", line 103, in get_backbone
        model = model_fn(*args, **kwargs)
      File "C:\Users\Giulia\PycharmProjects\multiclass_new\venv\lib\site- 
   packages\classification_models\models_factory.py", line 78, in wrapper
        return func(*args, **new_kwargs)
      File "C:\Users\Giulia\PycharmProjects\multiclass_new\venv\lib\site- 
   packages\classification_models\models\resnet.py", line 314, in ResNet34
        return ResNet(
      File "C:\Users\Giulia\PycharmProjects\multiclass_new\venv\lib\site- 
   packages\classification_models\models\resnet.py", line 280, in ResNet
        load_model_weights(model, model_params.model_name,
      File "C:\Users\Giulia\PycharmProjects\multiclass_new\venv\lib\site- 
   packages\classification_models\weights.py", line 25, in load_model_weights
        weights_path = keras_utils.get_file(
    AttributeError: module 'keras.utils' has no attribute 'get_file'

标签： pythonkeras

解决方案

你可以试试：

import segmentation_models as sm

sm.set_framework('tf.keras')

sm.framework()

在 google colab 上为我工作！

推荐阅读