diff --git a/detectron/datasets/json_dataset_evaluator.py b/detectron/datasets/json_dataset_evaluator.py index 7ed3ce42c..92223e9fe 100644 --- a/detectron/datasets/json_dataset_evaluator.py +++ b/detectron/datasets/json_dataset_evaluator.py @@ -55,8 +55,10 @@ def evaluate_masks( if json_dataset.name.find('test') == -1: coco_eval = _do_segmentation_eval(json_dataset, res_file, output_dir) else: - logger.warning('Segmentation eval was ignored as annotations are undisclosed on test: {} ignored' - .format(json_dataset.name)) + logger.warning( + '{} eval ignored as annotations are undisclosed on test: {} ignored' + .format("Segmentation", json_dataset.name) + ) coco_eval = None # Optionally cleanup results json file if cleanup: @@ -139,8 +141,10 @@ def evaluate_boxes( if json_dataset.name.find('test') == -1: coco_eval = _do_detection_eval(json_dataset, res_file, output_dir) else: - logger.warning('Bbox eval was ignored as annotations are undisclosed on test: {} ignored' - .format(json_dataset.name)) + logger.warning( + '{} eval ignored as annotations are undisclosed on test: {} ignored' + .format("Bbox", json_dataset.name) + ) coco_eval = None # Optionally cleanup results json file if cleanup: @@ -341,8 +345,10 @@ def evaluate_keypoints( if json_dataset.name.find('test') == -1: coco_eval = _do_keypoint_eval(json_dataset, res_file, output_dir) else: - logger.warning('Keypoints evaluation was ignored as annotations are undisclosed on test:{} ignored' - .format(json_dataset.name)) + logger.warning( + '{} eval ignored as annotations are undisclosed on test: {} ignored' + .format("Keypoints", json_dataset.name) + ) coco_eval = None # Optionally cleanup results json file if cleanup: