helper

Copyright (c) 2025 Sima Technologies, Inc.

SPDX-License-Identifier: Apache-2.0

Classes

YoloHelpers

Yolo Helper class

Functions

postprocess(→ numpy.ndarray)
load_threshold(path)	Helper function to load the threshold
load_quantile(path)	Helper function to load the quantile
load_mean_std(path)	Helper function to load the mean and standard deviation
merge_levels(→ numpy.ndarray)
effad_postproc(→ dict)

Module Contents

class helper.YoloHelpers

Yolo Helper class

static nms(boxes, scores, iou_threshold)

static bridge(image: numpy.ndarray, mask: numpy.ndarray, **kwds) → numpy.ndarray

static flip_image(image: numpy.array) → bool

static compute_iou(box, boxes)

static xywh2xyxy(x)

static sigmoid(x)

static rescale_boxes(boxes, input_shape, image_shape)

static extract_boxes(box_predictions, img_height, img_width)

static process_box_output(box_output, img_height=3024, img_width=4032, conf_thres=0.5, iou_thres=0.5, num_masks=32)

static process_mask_output(mask_predictions, mask_output, boxes, img_height, img_width)

helper.postprocess(outputs: List[numpy.ndarray], input_height, input_width) → numpy.ndarray

helper.load_threshold(path: pathlib.Path)

Helper function to load the threshold

helper.load_quantile(path: pathlib.Path)

Helper function to load the quantile

helper.load_mean_std(path: pathlib.Path)

Helper function to load the mean and standard deviation

helper.merge_levels(preds: List[numpy.ndarray]) → numpy.ndarray

helper.effad_postproc(yolo_output: dict, effad_output: dict, cls_threshold: float) → dict