correctly update region class stats in seg dataset

mittagessen · Feb 9, 2024 · e654cf6 · e654cf6
1 parent ef324e0
commit e654cf6
Showing 1 changed file with 1 addition and 0 deletions.
diff --git a/kraken/lib/dataset/segmentation.py b/kraken/lib/dataset/segmentation.py
@@ -151,6 +151,7 @@ def add(self, doc: 'Segmentation'):
                 if reg_type not in self.class_mapping['regions']:
                     self.num_classes += 1
                     self.class_mapping['regions'][reg_type] = self.num_classes - 1
+                self.class_stats['regions'][reg_type] += 1
         self.targets.append({'baselines': baselines_, 'regions': regions_})
         self.imgs.append(doc.imagename)