diff --git a/configs/ssd512_coco.py b/configs/ssd512_coco.py index 67fe0309c6bd3b05bf151011685dbd40ca732b91..a23a1b67dd4c20362df5d6daad34ae731ca12173 100644 --- a/configs/ssd512_coco.py +++ b/configs/ssd512_coco.py @@ -65,7 +65,7 @@ train_pipeline = [ type='MinIoURandomCrop', min_ious=(0.1, 0.3, 0.5, 0.7, 0.9), min_crop_size=0.3), - dict(type='Resize', img_scale=(300, 300), keep_ratio=False), + dict(type='Resize', img_scale=(512, 512), keep_ratio=False), dict(type='Normalize', **img_norm_cfg), dict(type='RandomFlip', flip_ratio=0.5), dict(type='DefaultFormatBundle'), @@ -75,7 +75,7 @@ test_pipeline = [ dict(type='LoadImageFromFile'), dict( type='MultiScaleFlipAug', - img_scale=(300, 300), + img_scale=(512, 512), flip=False, transforms=[ dict(type='Resize', keep_ratio=False), diff --git a/mmdet/datasets/pipelines/transforms.py b/mmdet/datasets/pipelines/transforms.py index 621be4d4afe2332b3392ea61f989637e6c18634e..18b1cc10a3c0153118aa36a3b917d405d46d86e5 100644 --- a/mmdet/datasets/pipelines/transforms.py +++ b/mmdet/datasets/pipelines/transforms.py @@ -527,7 +527,7 @@ class Expand(object): left = int(random.uniform(0, w * ratio - w)) top = int(random.uniform(0, h * ratio - h)) expand_img[top:top + h, left:left + w] = img - boxes = boxes + np.tile((left, top), 2) + boxes = boxes + np.tile((left, top), 2).astype(boxes.dtype) results['img'] = expand_img results['gt_bboxes'] = boxes