imagedetectionyolo_mapper

February 4, 2026 · View on GitHub

Perform object detection using YOLO on images and return bounding boxes and class labels.

This operator uses a YOLO model to detect objects in images. It processes each image in the sample, returning the bounding boxes and class labels for detected objects. The operator sets the bbox_tag and class_label_tag fields in the sample's metadata. If no image is present or no objects are detected, it sets bbox_tag to an empty array and class_label_tag to -1. The operator uses a confidence score threshold and IoU (Intersection over Union) score threshold to filter detections.

使用 YOLO 对图像进行目标检测并返回边界框和类别标签。

该算子使用 YOLO 模型检测图像中的目标。它处理样本中的每张图像,返回检测到的目标的边界框和类别标签。该算子在样本的元数据中设置 bbox_tagclass_label_tag 字段。如果没有图像或没有检测到目标,它将 bbox_tag 设置为空数组并将 class_label_tag 设置为 -1。该算子使用置信度分数阈值和 IoU(交并比)分数阈值来过滤检测结果。

Type 算子类型: mapper

Tags 标签: gpu, image

🔧 Parameter Configuration 参数配置

name 参数名type 类型default 默认值desc 说明
imgsz640resolution for image resizing
conf0.05confidence score threshold
iou0.5IoU (Intersection over Union) score threshold
model_path'yolo11n.pt'the path to the YOLO model.
args''
kwargs''