From ca905d05046ebacbc5dfaebb2ddf716cfe5fe667 Mon Sep 17 00:00:00 2001 From: weiyithu Date: Tue, 28 Mar 2023 23:55:57 +0800 Subject: [PATCH] fix typos in evaluation metrics --- .../datasets/evaluation_metrics.py | 16 ++++++++-------- 1 file changed, 8 insertions(+), 8 deletions(-) diff --git a/projects/mmdet3d_plugin/datasets/evaluation_metrics.py b/projects/mmdet3d_plugin/datasets/evaluation_metrics.py index 2ce40ce..1b0f356 100644 --- a/projects/mmdet3d_plugin/datasets/evaluation_metrics.py +++ b/projects/mmdet3d_plugin/datasets/evaluation_metrics.py @@ -58,15 +58,15 @@ def evaluation_semantic(pred_occ, gt_occ, img_metas, class_num): gt_i = gt_to_voxel(gt_i, img_metas) mask = (gt_i != 255) score = np.zeros((class_num, 3)) - for i in range(class_num): - if i == 0: #class 0 for geometry IoU - score[i][0] += ((gt_i[mask] != 0) * (pred_i[mask] != 0)).sum() - score[i][1] += (gt_i[mask] != 0).sum() - score[i][2] += (pred_i[mask] != 0).sum() + for j in range(class_num): + if j == 0: #class 0 for geometry IoU + score[j][0] += ((gt_i[mask] != 0) * (pred_i[mask] != 0)).sum() + score[j][1] += (gt_i[mask] != 0).sum() + score[j][2] += (pred_i[mask] != 0).sum() else: - score[i][0] += ((gt_i[mask] == i) * (pred_i[mask] == i)).sum() - score[i][1] += (gt_i[mask] == i).sum() - score[i][2] += (pred_i[mask] == i).sum() + score[j][0] += ((gt_i[mask] == j) * (pred_i[mask] == j)).sum() + score[j][1] += (gt_i[mask] == j).sum() + score[j][2] += (pred_i[mask] == j).sum() results.append(score) return np.stack(results, axis=0) \ No newline at end of file