fixed label mapping mistakes

reghbali · Sep 19, 2024 · f7183fb · f7183fb
1 parent 3e42291
commit f7183fb
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,