Skip to content
Snippets Groups Projects
Commit 7a05b93f authored by Bobholamovic's avatar Bobholamovic
Browse files

Fix holes in Lebedev dataset

parent 9126a3ce
No related branches found
No related tags found
1 merge request!2Update outdated code
......@@ -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
0% Loading or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment