Fix Bug: 日志中没有mIoU

main.py

@@ -215,7 +215,6 @@ class EvalCallBack(Callback):
         Iou = IOUMetric(6)
         if cur_epoch % self.eval_per_epoch == 0:
             acc = self.model.eval(self.eval_dataset, dataset_sink_mode=False)
-            self.epoch_per_eval["epoch"].append(cur_epoch)
             self.epoch_per_eval["acc"].append(acc["Accuracy"])
             print(acc)
             for data in self.eval_dataset.create_dict_iterator():
@@ -233,7 +232,7 @@ class EvalCallBack(Callback):
 time_cb = TimeMonitor(data_size=60)
 loss_cb = LossMonitor()
 model = Model(net, net_loss, net_opt, metrics={"Accuracy": Accuracy()})
-epoch_per_eval = {"epoch": [], "acc": [], "miou": []}
+epoch_per_eval = {"acc": [], "miou": []}
 eval_per_epoch = 5
 eval_cb = EvalCallBack(model, val_dataset, eval_per_epoch, epoch_per_eval)
 model.train(250, train_dataset, callbacks=[time_cb, loss_cb, ckpoint, eval_cb], dataset_sink_mode=False)