Commit 29135c22 authored by zhengmiao's avatar zhengmiao Committed by zhouzaida
Browse files

[Fix] Nearest interpolation to resize seg map

parent 39eb4ec6
...@@ -210,13 +210,13 @@ class Resize(BaseTransform): ...@@ -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): ...@@ -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
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment