Merge pull request vqdang#155 from zeeshanalipanhwar/patch-1

FIX: remove artifact merging
samjmolyneux · Aug 7, 2021 · 61ffeca · 61ffeca
2 parents a0f80c7 + 786d921
commit 61ffeca
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/dataloader/train_loader.py b/dataloader/train_loader.py
@@ -97,7 +97,7 @@ def __getitem__(self, idx):
         if self.with_type:
             type_map = (ann[..., 1]).copy()
             type_map = cropping_center(type_map, self.mask_shape)
-            type_map[type_map == 5] = 1  # merge neoplastic and non-neoplastic
+            #type_map[type_map == 5] = 1  # merge neoplastic and non-neoplastic
             feed_dict["tp_map"] = type_map
 
         # TODO: document hard coded assumption about #input