diff --git a/src/data/Lebedev.py b/src/data/Lebedev.py index bad5651ee7f37bdeebc2df780fee29050abc4c04..ecdf5ba62af3b9b035d89170f9bd49d24701df02 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