diff --git a/mmdet/apis/inference.py b/mmdet/apis/inference.py index a87323cee1aba6e97c75f2a563c5337bc5fe32ff..a39763a18bb108116c925e187b54600dd4ba1dcf 100644 --- a/mmdet/apis/inference.py +++ b/mmdet/apis/inference.py @@ -10,7 +10,9 @@ from mmdet.core import get_classes def _prepare_data(img, img_transform, cfg, device): ori_shape = img.shape img, img_shape, pad_shape, scale_factor = img_transform( - img, scale=cfg.data.test.img_scale) + img, + scale=cfg.data.test.img_scale, + keep_ratio=cfg.data.test.get('resize_keep_ratio', True)) img = to_tensor(img).to(device).unsqueeze(0) img_meta = [ dict(