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

Fix holes in Lebedev dataset

parent 9126a3ce
Branches
No related tags found
1 merge request!2Update outdated code
This commit is part of merge request !2. Comments created here will be created in the context of that merge request.
Hide whitespace changes
Inline Side-by-side
Showing
with 1 addition and 1 deletion
src/data/Lebedev.py
+ 1
− 1
Edit View file @ 7a05b93f
...@@ -44,4 +44,4 @@ class LebedevDataset(CDDataset): ...@@ -44,4 +44,4 @@ class LebedevDataset(CDDataset):
def fetch_label(self, label_path): def fetch_label(self, label_path):
# To {0,1} # To {0,1}
return (super().fetch_label(label_path) / 255.0).astype(np.uint8) return (super().fetch_label(label_path) > 127).astype(np.uint8)
\ No newline at end of file \ No newline at end of file
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Please register or to comment