def create_and_add_common_ground_truth(self): # Add groundtruth self.wp_eval = ( object_detection_evaluation.WeightedPascalDetectionEvaluator( self.categories)) image_key1 = 'img1' groundtruth_boxes1 = np.array( [[0, 0, 1, 1], [0, 0, 2, 2], [0, 0, 3, 3]], dtype=float) groundtruth_class_labels1 = np.array([1, 3, 1], dtype=int) self.wp_eval.add_single_ground_truth_image_info( image_key1, { standard_fields.InputDataFields.groundtruth_boxes: groundtruth_boxes1, standard_fields.InputDataFields.groundtruth_classes: groundtruth_class_labels1 }) # add 'img2' separately image_key3 = 'img3' groundtruth_boxes3 = np.array([[0, 0, 1, 1]], dtype=float) groundtruth_class_labels3 = np.array([2], dtype=int) self.wp_eval.add_single_ground_truth_image_info( image_key3, { standard_fields.InputDataFields.groundtruth_boxes: groundtruth_boxes3, standard_fields.InputDataFields.groundtruth_classes: groundtruth_class_labels3 })
def test_value_error_on_duplicate_images(self): # Add groundtruth self.wp_eval = ( object_detection_evaluation.WeightedPascalDetectionEvaluator( self.categories)) image_key1 = 'img1' groundtruth_boxes1 = np.array([[0, 0, 1, 1], [0, 0, 2, 2], [0, 0, 3, 3]], dtype=float) groundtruth_class_labels1 = np.array([1, 3, 1], dtype=int) self.wp_eval.add_single_ground_truth_image_info( image_key1, {standard_fields.InputDataFields.groundtruth_boxes: groundtruth_boxes1, standard_fields.InputDataFields.groundtruth_classes: groundtruth_class_labels1}) with self.assertRaises(ValueError): self.wp_eval.add_single_ground_truth_image_info( image_key1, {standard_fields.InputDataFields.groundtruth_boxes: groundtruth_boxes1, standard_fields.InputDataFields.groundtruth_classes: groundtruth_class_labels1})