From 29135c22ab3b35bd12d7bf9c9f00cdf18d8124e9 Mon Sep 17 00:00:00 2001 From: zhengmiao Date: Wed, 22 Jun 2022 07:50:41 +0000 Subject: [PATCH] [Fix] Nearest interpolation to resize seg map --- mmcv/transforms/processing.py | 6 +++--- 1 file changed, 3 insertions(+), 3 deletions(-) diff --git a/mmcv/transforms/processing.py b/mmcv/transforms/processing.py index 43d0cd890..a0749fa48 100644 --- a/mmcv/transforms/processing.py +++ b/mmcv/transforms/processing.py @@ -210,13 +210,13 @@ class Resize(BaseTransform): gt_seg = mmcv.imrescale( results['gt_seg_map'], results['scale'], - interpolation=self.interpolation, + interpolation='nearest', backend=self.backend) else: gt_seg = mmcv.imresize( results['gt_seg_map'], results['scale'], - interpolation=self.interpolation, + interpolation='nearest', backend=self.backend) results['gt_seg_map'] = gt_seg @@ -938,7 +938,7 @@ class RandomChoiceResize(BaseTransform): assert mmcv.is_list_of(self.scales, tuple) self.resize_cfg = dict(type=resize_type, **resize_kwargs) - # create a empty Reisize object + # create a empty Resize object self.resize = TRANSFORMS.build({'scale': 0, **self.resize_cfg}) @cache_randomness