neodroidvision.data.synthesis.conversion.mnist.obj_dect_gen

Created on {date}

Functions

calculate_iou(prediction_box, gt_box)

Calculate intersection over union of single predicted and ground truth box. :param prediction_box: location of predicted object as [xmin, ymin, xmax, ymax] :type prediction_box: numpy.array of floats :param gt_box: location of ground truth object as [xmin, ymin, xmax, ymax] :type gt_box: numpy.array of floats :param returns: float: value of the intersection of union for the two boxes.

compute_iou_all(bbox, all_bboxes)

param bbox

dataset_exists(dirpath, num_images)

param dirpath

download_mnist()

description

extract_mnist()

return

generate_dataset(dirpath, num_images, ...)

param dirpath

load()

return

tight_bbox(digit, orig_bbox)

param digit