uavfpy.odcl.utils.drawer

Module Contents

Classes

TargetDrawer

class uavfpy.odcl.utils.drawer.TargetDrawer(labels)

Bases: object

static get_rand_color()
draw_tile_frame(self, img, alpha=0.9)

draw a frame around the input. Useful for visualizing tiles.

draw_target_bbox(self, img, target, color=None)

Draw a bbox, class label, and confidence score around a target onto image

updated image with target drawn onto it

draw_all(self, img, targets, color=None)

Draw all current targets onto img

Parameters
imgcv2 Image

(H, W, 3) 8-bit

targets: list of Target

targets to draw

Returns
(H, W, 3) 8-bit image

Image with targets drawn

make_target_bbox_img_opencv(self, img, targets)