Skip to content

Commit 9287d29

Browse files
author
mykola
committed
Updated docs
1 parent b655a7a commit 9287d29

1 file changed

Lines changed: 3 additions & 2 deletions

File tree

docs/source/models/detectors/yolo_onnx_detector.md

Lines changed: 3 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -47,11 +47,12 @@ detected_df = detector.transform(input_df)
4747
| onlyRotated | bool | Return only rotated boxes | False |
4848
| model | str | Model identifier | (required) |
4949
| padding | int | Padding percent to expand detected boxes | 0 |
50+
| labels | list | List of class labels for detected objects | [] |
5051

5152
## Notes
5253
- The detector loads YOLO ONNX models from Hugging Face Hub or local path.
5354
- Supports batch and distributed processing with Spark.
5455
- Padding expands detected bounding boxes by a percentage.
56+
- The `labels` parameter allows you to specify custom class labels for the detected objects. If provided, detection results will use these labels instead of class indices.
5557
- Used as a base for specialized detectors (e.g., [**Face Detector**]
56-
(#FaceDetector), [**Signature Detector**](#SignatureDetector)).
57-
58+
(#FaceDetector), [**Signature Detector**](#SignatureDetector)).

0 commit comments

Comments
 (0)