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