0

对于物体检测,我使用的是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)作为prepossessing,然后再发送到预测器?或者 DefaultPredictor 在预测之前调整图像大小,我必须重写一个函数来调整大小为 (1200, 1200)?

4

1 回答 1

2

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

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

于 2021-02-16T20:51:48.010 回答