mirrors
/
mmcv
mirror of https://github.com/open-mmlab/mmcv.git
1
0
Fork 0
Code Issues Projects Releases Wiki Activity

[Fix] Nearest interpolation to resize seg map

pull/2133/head
zhengmiao 2022-06-22 07:50:41 +00:00 committed by zhouzaida
parent 39eb4ec62b
commit 29135c22ab
1 changed files with 3 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

6
mmcv/transforms/processing.py
View File

@ -210,13 +210,13 @@ class Resize(BaseTransform):
gt_seg = mmcv.imrescale( gt_seg = mmcv.imrescale(
results['gt_seg_map'], results['gt_seg_map'],
results['scale'], results['scale'],
interpolation=self.interpolation, interpolation='nearest',
backend=self.backend) backend=self.backend)
else: else:
gt_seg = mmcv.imresize( gt_seg = mmcv.imresize(
results['gt_seg_map'], results['gt_seg_map'],
results['scale'], results['scale'],
interpolation=self.interpolation, interpolation='nearest',
backend=self.backend) backend=self.backend)
results['gt_seg_map'] = gt_seg results['gt_seg_map'] = gt_seg
@ -938,7 +938,7 @@ class RandomChoiceResize(BaseTransform):
assert mmcv.is_list_of(self.scales, tuple) assert mmcv.is_list_of(self.scales, tuple)
self.resize_cfg = dict(type=resize_type, **resize_kwargs) 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}) self.resize = TRANSFORMS.build({'scale': 0, **self.resize_cfg})
@cache_randomness @cache_randomness