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