From 102f9884fd7a7f6f7183b66146b9ddabf61a568e Mon Sep 17 00:00:00 2001 From: wooqy Date: Thu, 20 Oct 2022 17:12:54 +1100 Subject: [PATCH] update --- code/datasets/scannet_dataset.py | 3 ++- 1 file changed, 2 insertions(+), 1 deletion(-) diff --git a/code/datasets/scannet_dataset.py b/code/datasets/scannet_dataset.py index 284e531..9831fd7 100644 --- a/code/datasets/scannet_dataset.py +++ b/code/datasets/scannet_dataset.py @@ -194,9 +194,10 @@ def load_meta_data(self, split_path, img_dir, seg_dir, pose_dir): ori_sems = cv2.resize(imageio.imread(os.path.join(seg_dir, "%s.png" % str(cur_id))), (self.img_res[1], self.img_res[0]), interpolation=cv2.INTER_NEAREST).\ reshape(1, -1).transpose(1, 0) + ins_list = np.unique(ori_sems) cur_sems = np.copy(ori_sems) if self.label_mapping is not None: - for i in self.label_mapping: + for i in ins_list if self.instance else self.label_mapping: # cur_sems[cur_sems == i] = self.label_mapping.index(i) cur_sems[ori_sems == i] = self.label_mapping.index(self.instance_mapping_dict[i]) if self.instance else self.label_mapping(i) split_sems.append(cur_sems)