fixed histogramming

shoaib42 · Feb 15, 2022 · f2dae77 · f2dae77
1 parent 480865c
commit f2dae77
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/python/examples/segnet_utils.py b/python/examples/segnet_utils.py
@@ -80,7 +80,7 @@ def ComputeStats(self):
         self.net.Mask(self.class_mask, self.grid_width, self.grid_height)
 
         # compute the number of times each class occurs in the mask
-        class_histogram, _ = np.histogram(self.class_mask_np, self.num_classes)
+        class_histogram, _ = np.histogram(self.class_mask_np, bins=self.num_classes, range=(0, self.num_classes-1))
 
         print('grid size:   {:d}x{:d}'.format(self.grid_width, self.grid_height))
         print('num classes: {:d}'.format(self.num_classes))