From f78b70e6aabfecda2d33164b9d3585f77f8d2645 Mon Sep 17 00:00:00 2001 From: Radoyeh Shojaei Date: Tue, 23 Mar 2021 12:43:06 -0700 Subject: [PATCH] Fix mask generation in base_loader when reading an image --- datasets/base_loader.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/datasets/base_loader.py b/datasets/base_loader.py index 8823884..73daf53 100644 --- a/datasets/base_loader.py +++ b/datasets/base_loader.py @@ -173,9 +173,9 @@ def read_images(self, img_path, mask_path, mask_out=False): if (mask_out): mask = self.drop_mask * mask - mask = mask.copy() + mask_copy = mask.copy() for k, v in self.id_to_trainid.items(): - binary_mask = (mask == k) #+ (gtCoarse == k) + binary_mask = (mask_copy == k) #+ (gtCoarse == k) if ('refinement' in mask_path) and cfg.DROPOUT_COARSE_BOOST_CLASSES != None and v in cfg.DROPOUT_COARSE_BOOST_CLASSES and binary_mask.sum() > 0 and 'vidseq' not in mask_path: binary_mask += (gtCoarse == k) binary_mask[binary_mask >= 1] = 1