Add class index in the object detection evaluation logging info.

PiperOrigin-RevId: 450994731

Add class index in the object detection evaluation logging info.
PiperOrigin-RevId: 450994731
2bd76d13 · Liangzhe Yuan · TF Object Detection Team · 5212bb9f · 2bd76d13
Commit 2bd76d13 authored May 25, 2022 by Liangzhe Yuan Committed by TF Object Detection Team May 25, 2022
Hide whitespace changes
Inline Side-by-side

Showing with 2 additions and 1 deletion

research/object_detection/utils/object_detection_evaluation.py ...rch/object_detection/utils/object_detection_evaluation.py +2 -1

No files found.
--- a/research/object_detection/utils/object_detection_evaluation.py
+++ b/research/object_detection/utils/object_detection_evaluation.py
@@ -1335,7 +1335,8 @@ class ObjectDetectionEvaluation(object):
      average_precision = metrics.compute_average_precision(
          precision_within_bound, recall_within_bound)
      self.average_precision_per_class[class_index] = average_precision
-      logging.info('average_precision: %f', average_precision)
+      logging.info(
+          'class %d average_precision: %f', class_index, average_precision)
    self.corloc_per_class = metrics.compute_cor_loc(
        self.num_gt_imgs_per_class,