Skip to content

GitLab

Commit 2bd76d13 authored by Liangzhe Yuan's avatar Liangzhe Yuan Committed by TF Object Detection Team
Browse files

Add class index in the object detection evaluation logging info. 

PiperOrigin-RevId: 450994731
parent 5212bb9f
Show whitespace changes
Inline Side-by-side
Showing with 2 additions and 1 deletion
research/object_detection/utils/object_detection_evaluation.py
View file @ 2bd76d13
...@@ -1335,7 +1335,8 @@ class ObjectDetectionEvaluation(object): ...@@ -1335,7 +1335,8 @@ class ObjectDetectionEvaluation(object):
average_precision = metrics.compute_average_precision( average_precision = metrics.compute_average_precision(
precision_within_bound, recall_within_bound) precision_within_bound, recall_within_bound)
self.average_precision_per_class[class_index] = average_precision 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.corloc_per_class = metrics.compute_cor_loc(
self.num_gt_imgs_per_class, self.num_gt_imgs_per_class,
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment