DefaultPredictor of Detectron2中的图像大小

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))
        ]))

对于这个预测，我想知道的是，在发送到预测器之前，我可以使用DefaultPredictor of detectron2并将图像大小调整为(1200,1200)。或者DefaultPredictor在预测之前调整图像大小，而我必须重写一个函数来调整大小为(1200,1200)？

Answer 1

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

DefaultPredictor应用ResizeShortestEdge转换(可以在配置文件中配置)，但这不是您想要的。