Skip to main content

Triton classifier

Classification via Triton Inference Server. Must chain from a parent model using operate_on.

models:
- name: "triton-detector"
type: "triton_detector"
triton:
model_name: "detector_model"
server_url: "localhost:8001"
label_file: /path/to/labels.txt

- name: "triton-classifier"
type: "triton_classifier"
triton:
model_name: "classifier_model"
server_url: "localhost:8001"
label_file: /path/to/class_labels.txt
operate_on: "triton-detector"
operate_on_classes: [0, 2]

Fields

FieldTypeRequiredDefaultDescription
tritonobjectYesTriton server configuration (same fields as Triton detector)
label_filestringYesPath to label file
operate_onstringYesName of parent model
operate_on_classeslist of intNoall parent classesParent class IDs to classify