pipeline/detectors/__init__.py

from .yolo import YOLO


def build_detector(config):
    """
    Build the detection model based on the provided configuration.
    """
    # Initialize the YOLO object detection model
    return YOLO(config.thresholds.confidence,
                config.thresholds.iou,
                config.slicing.overlap,
                config.categories,
                config.device)