fix _resize_seg bug in Resize data transforms (#246)

This commit is contained in:
Junjun2016 2020-11-16 17:28:41 +08:00 committed by GitHub
parent 2bd51e60fd
commit 300f42d006
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23

View File

@ -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