neodroidvision.data.synthesis.conversion.mnist.obj_dect_gen.calculate_iou

neodroidvision.data.synthesis.conversion.mnist.obj_dect_gen.calculate_iou(prediction_box, gt_box)[source]

Calculate intersection over union of single predicted and ground truth box. :param prediction_box: location of predicted object as

[xmin, ymin, xmax, ymax]

Parameters
  • gt_box (numpy.array of python:floats) – location of ground truth object as [xmin, ymin, xmax, ymax]

  • returns – float: value of the intersection of union for the two boxes.