Skip to content

modules.vision.detection.utils

Functions

get_class_colour(cls_num: int)

Generate a unique color for each class ID.

draw_detections(frame, results, conf_threshold = 0.4)

Draw YOLOv8 detection boxes and labels on a frame.