Update train_loader.py

Commented out the line 100 that merges the inst_type==5 with the first.
samjmolyneux · Aug 7, 2021 · 786d921 · 786d921
1 parent a0f80c7
commit 786d921
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