我遵循教程的指示,以便为灰度图像上的1类创建对象检测器。下面是我的代码(请注意,我在自己的数据集中使用DenseNet作为主干预培训模型):
# put the pieces together inside a FasterRCNN体系结构,我认为transform组件中存在问题,因为它试图规范化最初是灰度的图像,而
然而,它说'FasterRCNN‘对象没有属性’特性‘--我想在它有36个类时提取( 36,2048)形状特征。是否有任何方法可以用预先训练过的吡炬模型来提取。model = torchvision.models.detection.fasterrcnn_resnet50_fpn(pretrained=True).to(device)
features = list