Merge pull request #57 from reghbali/reghbali_synthseg_label_fix

fixed label mapping mistakes
reghbali · Sep 20, 2024 · 9d7204b · 9d7204b
2 parents 3e42291 + f7183fb
commit 9d7204b
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/pyalfe/tasks/segmentation.py b/pyalfe/tasks/segmentation.py
@@ -419,8 +419,8 @@ def post_process(self, pred, seg, mask=None):
                 26: 4,
                 28: 4,
                 49: 4,
-                50: 41,
-                51: 41,
+                50: 4,
+                51: 4,
                 52: 4,
                 54: 4,
                 58: 4,