为了创建我自己的对象检测,我想完善火炬视觉中可用的经过预先训练的RetinaNet模型。我试图复制在这个链接上为FastRCNN所做的工作:https://pytorch.org/tutorials/intermediate/torchvision_tutorial.html#finetuning-from-a-pretrained-model我所做的是:
model = model = torchvision<
我想提取ResNet101中的特性,但是导入torchvision.models.feature_extraction时遇到了问题。这是我的代码:from torchvision.models.feature_extractor import create_feature_extractor"layer2", "la