From 7a05b93fe522221c92d4a6f08750fd3cfd60b62a Mon Sep 17 00:00:00 2001
From: Bobholamovic <bob1998425@hotmail.com>
Date: Sun, 17 May 2020 12:11:48 +0800
Subject: [PATCH] Fix holes in Lebedev dataset

---
 src/data/Lebedev.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/src/data/Lebedev.py b/src/data/Lebedev.py
index bad5651..ecdf5ba 100644
--- a/src/data/Lebedev.py
+++ b/src/data/Lebedev.py
@@ -44,4 +44,4 @@ class LebedevDataset(CDDataset):
 
     def fetch_label(self, label_path):
         # To {0,1}
-        return (super().fetch_label(label_path) / 255.0).astype(np.uint8)  
\ No newline at end of file
+        return (super().fetch_label(label_path) > 127).astype(np.uint8)  
\ No newline at end of file
-- 
GitLab