Detectron2 的 DefaultPredictor 中的图像大小

Question

对于对象检测，我使用的是 detectron2。我想修复输入图像大小，所以我制作了自定义数据加载器：

def build_train_loader(cls, cfg):
    dataloader = build_detection_train_loader(cfg,
        mapper=DatasetMapper(cfg, is_train=True, augmentations=[
            T.Resize((1200, 1200))
        ]))

我想知道的是预测，我可以使用 detectron2 的 DefaultPredictor 并将我的图像大小调整为 (1200, 1200) 作为预处理，然后再发送到预测器？或者 DefaultPredictor 在预测之前调整图像大小，我必须覆盖一个函数以调整大小为 (1200, 1200)?

Answer 1

您必须自己预处理图像或编写自己的预测器，在调用模型之前应用调整大小。

DefaultPredictor applies a ResizeShortestEdge 转换（可以在配置文件中配置），但这并不是您想要的。

Detectron2 的 DefaultPredictor 中的图像大小

Image size in DefaultPredictor of Detectron2

pytorch

detectron