reformat eval scripts

bcb9d628 · zhangwenwei · 852295ce · bcb9d628
Commit bcb9d628 authored Apr 14, 2020 by zhangwenwei
Show whitespace changes
Inline Side-by-side

Showing with 8 additions and 11 deletions

mmdet3d/core/evaluation/kitti_utils/eval.py mmdet3d/core/evaluation/kitti_utils/eval.py +8 -11

No files found.
--- a/mmdet3d/core/evaluation/kitti_utils/eval.py
+++ b/mmdet3d/core/evaluation/kitti_utils/eval.py
@@ -687,21 +687,18 @@ def kitti_eval(gt_annos,
            result += ('{} AP@{:.2f}, {:.2f}, {:.2f}:\n'.format(
                curcls_name, *min_overlaps[i, :, j]))
            if mAPbbox is not None:
-                result += (
+                result += 'bbox AP:{:.4f}, {:.4f}, {:.4f}\n'.format(
-                    'bbox AP:{:.4f}, {:.4f}, {:.4f}\n'.format(*mAPbbox[j, :,
+                    *mAPbbox[j, :, i])
-                                                                       i]))
            if mAPbev is not None:
-                result += (
+                result += 'bev  AP:{:.4f}, {:.4f}, {:.4f}\n'.format(
-                    'bev  AP:{:.4f}, {:.4f}, {:.4f}\n'.format(*mAPbev[j, :,
+                    *mAPbev[j, :, i])
-                                                                      i]))
            if mAP3d is not None:
-                result += (
+                result += '3d   AP:{:.4f}, {:.4f}, {:.4f}\n'.format(
-                    '3d   AP:{:.4f}, {:.4f}, {:.4f}\n'.format(*mAP3d[j, :, i]))
+                    *mAP3d[j, :, i])
            if compute_aos:
-                result += (
+                result += 'aos  AP:{:.2f}, {:.2f}, {:.2f}\n'.format(
-                    'aos  AP:{:.2f}, {:.2f}, {:.2f}\n'.format(*mAPaos[j, :,
+                    *mAPaos[j, :, i])
-                                                                      i]))
            # prepare results for logger
            for idx in range(3):