我想在ResNet101中提取特征，但是，我在导入torchvision. models. feature_extraction时遇到了麻烦。
下面是我的代码：
from torchvision import models
from torchvision.models.feature_extractor import create_feature_extractor
res101 = models.resnet101(pretrained=True)
extractor = create_feature_extractor(
res101,
return_nodes=[
"conv1",
"maxpool",
"layer1",
"layer2",
"layer3",
"layer4",
]
)
features = extractor(inputs)这就是错误
from torchvision.models.feature_extractor import create_feature_extractor
Traceback (most recent call last):
Input In [11] in <cell line: 1>
from torchvision.models.feature_extractor import create_feature_extractor
ModuleNotFoundError: No module named 'torchvision.models.feature_extractor'
3条答案
按热度按时间cwdobuhd1#
您可能会尝试使用类似于以下内容的内容：
参见
extraction与extractor的对比Check this module
b91juud32#
同样的问题。我使用
conda安装了PyTorch，它在Jupyter笔记本电脑中工作正常。但是它在终端中不工作。原来
pip列出的torchvision版本是0.82。通过使用
pip更新torchvision解决。也许有些软件包帮我安装了旧版本，希望我的经验能帮到你。
t1qtbnec3#
卸载torch、torchvision和torchaudio，然后通过以下链接从预览版（每晚）安装这些包https://pytorch.org/get-started/locally/