From 300f42d0064ba223db967a37bc2aedcc778a7a60 Mon Sep 17 00:00:00 2001 From: Junjun2016 Date: Mon, 16 Nov 2020 17:28:41 +0800 Subject: [PATCH] fix _resize_seg bug in Resize data transforms (#246) --- mmseg/datasets/pipelines/transforms.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/mmseg/datasets/pipelines/transforms.py b/mmseg/datasets/pipelines/transforms.py index 2f4262fda..74be564c5 100644 --- a/mmseg/datasets/pipelines/transforms.py +++ b/mmseg/datasets/pipelines/transforms.py @@ -195,7 +195,7 @@ class Resize(object): else: gt_seg = mmcv.imresize( results[key], results['scale'], interpolation='nearest') - results['gt_semantic_seg'] = gt_seg + results[key] = gt_seg def __call__(self, results): """Call function to resize images, bounding boxes, masks, semantic