Detectron2 模型不产生任何结果答案

【问题标题】：Detectron2 models not generating any resultsDetectron2 模型不产生任何结果
【发布时间】：2022-02-04 02:03:21
【问题描述】：

我只是用下面的一些基本代码尝试detectron2

model = model_zoo.get('COCO-Detection/faster_rcnn_R_50_FPN_3x.yaml', trained=True)
im = Image.open('input.jpg')

t = transforms.ToTensor()
model.eval()

with torch.no_grad():
    im = t(im)
    output = model([{'image':im}])
    print(output)

但是模型没有产生任何有意义的预测

[{'instances': Instances(num_instances=0, image_height=480, image_width=640, fields=[pred_boxes: Boxes(tensor([], device='cuda:0', size=(0, 4))), scores: tensor([], device='cuda:0'), pred_classes: tensor([], device='cuda:0', dtype=torch.int64)])}]

我不太明白出了什么问题，detectron2 documentation 中指出：

你也可以像这样直接运行推理：
model.eval() 
with torch.no_grad():
   outputs = model(inputs) 

和

对于内置模型的推断，只需要“image”键，“width/height”是可选的。

在这种情况下，我似乎无法在此处找到缺失的链接。

【问题讨论】：

标签： deep-learning pytorch detectron

【解决方案1】：

我遇到了同样的问题，对我来说我有两个问题需要解决。第一个是调整最短边的大小。我使用了 Detectron2.data.transforms 中的 Detectron2 构建函数并导入了 ResizeShortestEdge。可以使用 cfg.INPUT 找到模型值，它将列出测试和训练的最大/最小尺寸。另一个问题是将颜色通道与 cfg 匹配。

【讨论】：