查看miou计算函数的中间步骤

miou.py

@@ -11,6 +11,8 @@ class IOUMetric:
         self.hist = np.zeros((num_classes, num_classes))
 
     def _fast_hist(self, label_pred, label_true):
+        print(label_pred.shape)
+        print(label_true.shape)
         mask = np.bitwise_and(label_true >= 0, label_true < self.num_classes)
         print(label_true[mask].shape)
         print(label_pred[mask].shape)