修改pred_mask

main.py

@@ -221,7 +221,7 @@ class EvalCallBack(Callback):
             for data in self.eval_dataset.create_dict_iterator():
                 image = Tensor(data['data'])
                 net_out = self.model.predict(image)
-                net_out = net_out.asnumpy()
+                net_out = np.amax(net_out.asnumpy(), axis=0)
                 Iou.add_batch(net_out, data['label'].asnumpy())
             acc_, acc_cls, iu, mean_iu, fwavacc = Iou.evaluate()
             print(acc_)

miou.py

@@ -11,11 +11,7 @@ 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)
         hist = np.bincount(
             self.num_classes * label_true[mask].astype(int) +
             label_pred[mask], minlength=self.num_classes ** 2).reshape(self.num_classes, self.num_classes)