Skip to content

Commit 61ffeca

Browse files
authored
Merge pull request vqdang#155 from zeeshanalipanhwar/patch-1
FIX: remove artifact merging
2 parents a0f80c7 + 786d921 commit 61ffeca

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

dataloader/train_loader.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -97,7 +97,7 @@ def __getitem__(self, idx):
9797
if self.with_type:
9898
type_map = (ann[..., 1]).copy()
9999
type_map = cropping_center(type_map, self.mask_shape)
100-
type_map[type_map == 5] = 1 # merge neoplastic and non-neoplastic
100+
#type_map[type_map == 5] = 1 # merge neoplastic and non-neoplastic
101101
feed_dict["tp_map"] = type_map
102102

103103
# TODO: document hard coded assumption about #input

0 commit comments

Comments
 (0)