From ff2f1f3c3213dee29ee7d5f08e4e69bf6271995f Mon Sep 17 00:00:00 2001
From: Cao Yuhang <yhcao6@gmail.com>
Date: Thu, 29 Aug 2019 15:24:02 +0800
Subject: [PATCH] Fix ssd bug (#1288)
* fix bug of ssd
* clean
---
configs/ssd512_coco.py | 4 ++--
mmdet/datasets/pipelines/transforms.py | 2 +-
2 files changed, 3 insertions(+), 3 deletions(-)
diff --git a/configs/ssd512_coco.py b/configs/ssd512_coco.py
index 67fe030..a23a1b6 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 621be4d..18b1cc1 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
--
GitLab