Skip to main content

Classifier (Model Chaining)

Classifiers run on objects detected by a parent model. They must reference a previously defined model using operate_on.

models:
- name: "vehicle-detector"
type: "detector"
architecture: "yolo"
onnx_file: /path/to/vehicle_model.onnx
label_file: /path/to/vehicle_labels.txt

- name: "color-classifier"
type: "classifier"
onnx_file: /path/to/color_model.onnx
label_file: /path/to/color_labels.txt
operate_on: "vehicle-detector"
operate_on_classes: [0, 2]
FieldDescription
operate_onRequired. Name of the parent detector. Must appear earlier in the list.
operate_on_classesOptional. Only classify objects of these class IDs from the parent. If omitted, all parent detections are classified.