random_bbox[source]

random_bbox(max_size=224)

bbox = random_bbox()
print(bbox)

tensor([ 4, 32, 62, 91])

visualize_bbox[source]

visualize_bbox(img, bbox, class_name, color=(255, 0, 0), thickness=2)

Visualizes a single bounding box on the image

visualize[source]

visualize(image, bboxes, category_ids, category_id_to_name, ax=None)

visualize(
    np.zeros((224,224,3)),
    [bbox],
    [0],
    {0:'BOX'},
)

Clipping input data to the valid range for imshow with RGB data ([0..1] for floats or [0..255] for integers).

plot_images[source]

plot_images(images)

images = [Image.open(path) for path in glob.glob('./data/samples/*.jpg')]
plot_images(images)