ModuleNotFoundError：没有名为“classification_models.resnet”的模块

Question

from classification_models.resnet import ResNet18, preprocess_input
model = ResNet18((224, 224, 3), weights='imagenet')

ModuleNotFoundError                       Traceback (most recent call last)
<ipython-input-43-ff8b34ae99fa> in <module>()
----> 1 from classification_models.resnet import ResNet18, preprocess_input
      2 
      3 model = ResNet18((224, 224, 3), weights='imagenet')

ModuleNotFoundError: No module named 'classification_models.resnet'

我无法在 google colab 上导入预训练的 ResNet18 模型。请帮忙

Answer 1

您可以访问所有ResNet (ResNet50, ResNet101, ResNet152) 模型及其 preprocess_input() 可以轻松地从tf.keras.applications API如下：

from tensorflow.keras.applications.resnet50 import ResNet50
from tensorflow.keras.applications.resnet50 import preprocess_input

请检查此链接了解有关如何使用和应用ResNet模型的更多详细信息。