Skip to main content

Active Classes

Use active_classes on a detector to only detect specific classes, ignoring all others. This is useful when your model outputs many classes but you only care about a few.

models:
- name: "detector"
type: "detector"
architecture: "yolo"
onnx_file: /path/to/model.onnx
label_file: /path/to/labels.txt
active_classes: [0, 2] # only detect class 0 and class 2

Class IDs correspond to line numbers (0-indexed) in the label file.