AttributeError: 模块'keras.utils' 没有使用classification_models.keras 的属性'get_file'

Question

当我尝试在我的计算机或 Google Colab 上运行下面的简单代码片段时：

from classification_models.keras import Classifiers

ResNet18, preprocess_input = Classifiers.get('resnet18')
resnet = ResNet18((170, 170, 3), weights='imagenet', include_top=False)

我收到以下错误：

AttributeError                            Traceback (most recent call last)
<ipython-input-4-b208d68b42cf> in <module>()
      2 
      3 ResNet18, preprocess_input = Classifiers.get('resnet18')
----> 4 resnet = ResNet18((170, 170, 3), weights='imagenet', include_top=False)

3 frames
/usr/local/lib/python3.7/dist-packages/classification_models/weights.py in load_model_weights(model, model_name, dataset, `classes`, include_top, **kwargs)
     23                              ' as true, classes should be {}'.format(weights['classes']))
     24 
---> 25         weights_path = keras_utils.get_file(
     26             weights['name'],
     27             weights['url'],

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

知道为什么吗？ 先感谢您！

Answer 1

好像是最新一期的包。 尽管如此，在本文档中，它表示如果您不提供任何文件路径，则weights默认为imagenet 。 因此，您可以尝试删除该参数，它应该可以工作。 请尝试：

from classification_models.keras import Classifiers

ResNet18, preprocess_input = Classifiers.get('resnet18')
resnet = ResNet18((170, 170, 3), include_top=False)