remove unused code in eg1800.py.

PaddlePaddle · Jul 13, 2021 · d417308 · d417308
1 parent c0a2cd9
commit d417308
Showing 1 changed file with 0 additions and 2 deletions.
diff --git a/paddleseg/datasets/eg1800.py b/paddleseg/datasets/eg1800.py
@@ -107,11 +107,9 @@ def __getitem__(self, item):
         label[label >= 0.5] = 1
         label[label < 0.5] = 0
 
-        # edge = show_edge(label)
         edge_mask = F.mask_to_binary_edge(
             label, radius=4, num_classes=self.num_classes)
         edge_mask = np.transpose(edge_mask, [1, 2, 0]).squeeze(axis=-1)
-        cv2.imwrite("edge.png", edge_mask * 255)
         im = np.concatenate([im_aug, im])
         if self.mode == "train":
             return im, label, edge_mask