From e60d34af8b0ec8f6e6439a0f3fb687042e9dde82 Mon Sep 17 00:00:00 2001
From: Cao Yuhang <yhcao6@gmail.com>
Date: Tue, 21 Jan 2020 16:30:05 +0800
Subject: [PATCH] Fix albu mask format bug (#1818)
---
mmdet/datasets/pipelines/transforms.py | 5 ++---
1 file changed, 2 insertions(+), 3 deletions(-)
diff --git a/mmdet/datasets/pipelines/transforms.py b/mmdet/datasets/pipelines/transforms.py
index 702bdea..a9516d5 100644
--- a/mmdet/datasets/pipelines/transforms.py
+++ b/mmdet/datasets/pipelines/transforms.py
@@ -833,9 +833,8 @@ class Albu(object):
results[label] = np.array(
[results[label][i] for i in results['idx_mapper']])
if 'masks' in results:
- results['masks'] = [
- results['masks'][i] for i in results['idx_mapper']
- ]
+ results['masks'] = np.array(
+ [results['masks'][i] for i in results['idx_mapper']])
if (not len(results['idx_mapper'])
and self.skip_img_without_anno):
--
GitLab