PiSAR / pipeline /detectors /__init__.py
eadali's picture
Test yolo onnx model
5f1587b
raw
history blame contribute delete
380 Bytes
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)