yolov7_postproc_overlay

Attributes

plugin_name
CLASSNAMES
COLORS

Classes

PostProcOpt
MyPlugin

Module Contents

yolov7_postproc_overlay.plugin_name = 'python_plugin'

yolov7_postproc_overlay.CLASSNAMES = ['person', 'bicycle', 'car', 'motorcycle', 'airplane', 'bus', 'train', 'truck', 'boat', 'traffic...

yolov7_postproc_overlay.COLORS

class yolov7_postproc_overlay.PostProcOpt(plugin, confidence_thres, iou_thres)

plugin

confidence_thres

iou_thres

grid_cells

final_buffer

post_process_optimized(model_output)

class yolov7_postproc_overlay.MyPlugin

MODEL_WIDTH = 640

MODEL_HEIGHT = 640

FRAME_WIDTH = 1280

FRAME_HEIGHT = 720

classes = 80

color_palette

confidence_thres = 0.25

iou_thres = 0.65

boxes = []

scores = []

class_ids = []

out_size = 0

frame_id = 0

model_outs = [(1, 80, 80, 6), (1, 80, 80, 6), (1, 80, 80, 243), (1, 40, 40, 6), (1, 40, 40, 6), (1, 40, 40,...

post_proc_opt

compute_iou(box, boxes)

nms(boxes, scores, iou_threshold)

rescale_boxes(boxes, input_width, input_height, target_width, target_height)

xywh2xyxy(x)

extract_boxes(boxes)

get_model_outputs(input_buffer)

run(input_buffers: List[python_plugin_template.SimaaiPythonBuffer], output_buffer: bytes) → None