robotic_sdk.examples package
Submodules
robotic_sdk.examples.apples_segmentation module
- class robotic_sdk.examples.apples_segmentation.YoloProcessor(model_path, conf=0.5)
Bases:
object- __init__(model_path, conf=0.5)
Initializes the YOLO model.
- Parameters:
model_path – Path to the YOLO model file.
conf – Confidence threshold for detections.
- process_frame(frame)
Processes a single frame and returns the annotated frame.
- Parameters:
frame – Input image as a NumPy array (BGR format).
- Returns:
Annotated image as a NumPy array (BGR format).
- robotic_sdk.examples.apples_segmentation.get_colours(cls_num)
Function to get class colors based on class number.