From 0372b3e86d693d3be68a4b2ff06c05ebe370542a Mon Sep 17 00:00:00 2001 From: daemon Date: Mon, 8 Jun 2020 20:16:48 +0700 Subject: [PATCH] Update transforms.py (#2932) * Update transforms.py * format with yapf Co-authored-by: Kai Chen --- mmdet/datasets/pipelines/transforms.py | 3 +-- 1 file changed, 1 insertion(+), 2 deletions(-) diff --git a/mmdet/datasets/pipelines/transforms.py b/mmdet/datasets/pipelines/transforms.py index b2203333b..d22ff1be4 100644 --- a/mmdet/datasets/pipelines/transforms.py +++ b/mmdet/datasets/pipelines/transforms.py @@ -294,8 +294,7 @@ class Pad(object): def _pad_masks(self, results): pad_shape = results['pad_shape'][:2] for key in results.get('mask_fields', []): - results[key] = results[key].pad( - pad_shape[:2], pad_val=self.pad_val) + results[key] = results[key].pad(pad_shape, pad_val=self.pad_val) def _pad_seg(self, results): for key in results.get('seg_fields', []):