Struct DecoderBuilder 

pub struct DecoderBuilder { /* private fields */ }

Implementations

impl DecoderBuilder

pub fn new() -> Self

Creates a default DecoderBuilder with no configuration and 0.5 score threshold and 0.5 OU threshold.

A valid confguration must be provided before building the Decoder.

Examples

let decoder = DecoderBuilder::new()
    .with_config_yaml_str(config_yaml)
    .build()?;
assert_eq!(decoder.score_threshold, 0.5);
assert_eq!(decoder.iou_threshold, 0.5);

pub fn with_config_yaml_str(self, yaml_str: String) -> Self

Loads a model configuration in YAML format. Does not check if the string is a correct configuration file. Use DecoderBuilder.build() to deserialize the YAML and parse the model configuration.

Examples

let config_yaml = include_str!("../../../testdata/modelpack_split.yaml").to_string();
let decoder = DecoderBuilder::new()
    .with_config_yaml_str(config_yaml)
    .build()?;

pub fn with_config_json_str(self, json_str: String) -> Self

Loads a model configuration in JSON format. Does not check if the string is a correct configuration file. Use DecoderBuilder.build() to deserialize the JSON and parse the model configuration.

Examples

let config_json = include_str!("../../../testdata/modelpack_split.json").to_string();
let decoder = DecoderBuilder::new()
    .with_config_json_str(config_json)
    .build()?;

pub fn with_config(self, config: ConfigOutputs) -> Self

Loads a model configuration. Does not check if the configuration is correct. Intended to be used when the user needs control over the deserialize of the configuration information. Use DecoderBuilder.build() to parse the model configuration.

Examples

let config_json = include_str!("../../../testdata/modelpack_split.json");
let config = serde_json::from_str(config_json)?;
let decoder = DecoderBuilder::new().with_config(config).build()?;

pub fn with_config_yolo_det( self, boxes: Detection, version: Option<DecoderVersion>, ) -> Self

Loads a YOLO detection model configuration. Use DecoderBuilder.build() to parse the model configuration.

Examples

let decoder = DecoderBuilder::new()
    .with_config_yolo_det(
        configs::Detection {
            anchors: None,
            decoder: configs::DecoderType::Ultralytics,
            quantization: Some(configs::QuantTuple(0.012345, 26)),
            shape: vec![1, 84, 8400],
            dshape: Vec::new(),
            normalized: Some(true),
        },
        None,
    )
    .build()?;

pub fn with_config_yolo_split_det(self, boxes: Boxes, scores: Scores) -> Self

Loads a YOLO split detection model configuration. Use DecoderBuilder.build() to parse the model configuration.

Examples

let boxes_config = configs::Boxes {
    decoder: configs::DecoderType::Ultralytics,
    quantization: Some(configs::QuantTuple(0.012345, 26)),
    shape: vec![1, 4, 8400],
    dshape: Vec::new(),
    normalized: Some(true),
};
let scores_config = configs::Scores {
    decoder: configs::DecoderType::Ultralytics,
    quantization: Some(configs::QuantTuple(0.0064123, -31)),
    shape: vec![1, 80, 8400],
    dshape: Vec::new(),
};
let decoder = DecoderBuilder::new()
    .with_config_yolo_split_det(boxes_config, scores_config)
    .build()?;

pub fn with_config_yolo_segdet( self, boxes: Detection, protos: Protos, version: Option<DecoderVersion>, ) -> Self

Loads a YOLO segmentation model configuration. Use DecoderBuilder.build() to parse the model configuration.

Examples

let seg_config = configs::Detection {
    decoder: configs::DecoderType::Ultralytics,
    quantization: Some(configs::QuantTuple(0.012345, 26)),
    shape: vec![1, 116, 8400],
    anchors: None,
    dshape: Vec::new(),
    normalized: Some(true),
};
let protos_config = configs::Protos {
    decoder: configs::DecoderType::Ultralytics,
    quantization: Some(configs::QuantTuple(0.0064123, -31)),
    shape: vec![1, 160, 160, 32],
    dshape: Vec::new(),
};
let decoder = DecoderBuilder::new()
    .with_config_yolo_segdet(
        seg_config,
        protos_config,
        Some(configs::DecoderVersion::Yolov8),
    )
    .build()?;

pub fn with_config_yolo_split_segdet( self, boxes: Boxes, scores: Scores, mask_coefficients: MaskCoefficients, protos: Protos, ) -> Self

Loads a YOLO split segmentation model configuration. Use DecoderBuilder.build() to parse the model configuration.

Examples

let boxes_config = configs::Boxes {
    decoder: configs::DecoderType::Ultralytics,
    quantization: Some(configs::QuantTuple(0.012345, 26)),
    shape: vec![1, 4, 8400],
    dshape: Vec::new(),
    normalized: Some(true),
};
let scores_config = configs::Scores {
    decoder: configs::DecoderType::Ultralytics,
    quantization: Some(configs::QuantTuple(0.012345, 14)),
    shape: vec![1, 80, 8400],
    dshape: Vec::new(),
};
let mask_config = configs::MaskCoefficients {
    decoder: configs::DecoderType::Ultralytics,
    quantization: Some(configs::QuantTuple(0.0064123, 125)),
    shape: vec![1, 32, 8400],
    dshape: Vec::new(),
};
let protos_config = configs::Protos {
    decoder: configs::DecoderType::Ultralytics,
    quantization: Some(configs::QuantTuple(0.0064123, -31)),
    shape: vec![1, 160, 160, 32],
    dshape: Vec::new(),
};
let decoder = DecoderBuilder::new()
    .with_config_yolo_split_segdet(boxes_config, scores_config, mask_config, protos_config)
    .build()?;

pub fn with_config_modelpack_det(self, boxes: Boxes, scores: Scores) -> Self

Loads a ModelPack detection model configuration. Use DecoderBuilder.build() to parse the model configuration.

Examples

let boxes_config = configs::Boxes {
    decoder: configs::DecoderType::ModelPack,
    quantization: Some(configs::QuantTuple(0.012345, 26)),
    shape: vec![1, 8400, 1, 4],
    dshape: Vec::new(),
    normalized: Some(true),
};
let scores_config = configs::Scores {
    decoder: configs::DecoderType::ModelPack,
    quantization: Some(configs::QuantTuple(0.0064123, -31)),
    shape: vec![1, 8400, 3],
    dshape: Vec::new(),
};
let decoder = DecoderBuilder::new()
    .with_config_modelpack_det(boxes_config, scores_config)
    .build()?;

pub fn with_config_modelpack_det_split(self, boxes: Vec<Detection>) -> Self

Loads a ModelPack split detection model configuration. Use DecoderBuilder.build() to parse the model configuration.

Examples

let config0 = configs::Detection {
    anchors: Some(vec![
        [0.13750000298023224, 0.2074074000120163],
        [0.2541666626930237, 0.21481481194496155],
        [0.23125000298023224, 0.35185185074806213],
    ]),
    decoder: configs::DecoderType::ModelPack,
    quantization: Some(configs::QuantTuple(0.012345, 26)),
    shape: vec![1, 17, 30, 18],
    dshape: Vec::new(),
    normalized: Some(true),
};
let config1 = configs::Detection {
    anchors: Some(vec![
        [0.36666667461395264, 0.31481480598449707],
        [0.38749998807907104, 0.4740740656852722],
        [0.5333333611488342, 0.644444465637207],
    ]),
    decoder: configs::DecoderType::ModelPack,
    quantization: Some(configs::QuantTuple(0.0064123, -31)),
    shape: vec![1, 9, 15, 18],
    dshape: Vec::new(),
    normalized: Some(true),
};

let decoder = DecoderBuilder::new()
    .with_config_modelpack_det_split(vec![config0, config1])
    .build()?;

pub fn with_config_modelpack_segdet( self, boxes: Boxes, scores: Scores, segmentation: Segmentation, ) -> Self

Loads a ModelPack segmentation detection model configuration. Use DecoderBuilder.build() to parse the model configuration.

Examples

let boxes_config = configs::Boxes {
    decoder: configs::DecoderType::ModelPack,
    quantization: Some(configs::QuantTuple(0.012345, 26)),
    shape: vec![1, 8400, 1, 4],
    dshape: Vec::new(),
    normalized: Some(true),
};
let scores_config = configs::Scores {
    decoder: configs::DecoderType::ModelPack,
    quantization: Some(configs::QuantTuple(0.0064123, -31)),
    shape: vec![1, 8400, 2],
    dshape: Vec::new(),
};
let seg_config = configs::Segmentation {
    decoder: configs::DecoderType::ModelPack,
    quantization: Some(configs::QuantTuple(0.0064123, -31)),
    shape: vec![1, 640, 640, 3],
    dshape: Vec::new(),
};
let decoder = DecoderBuilder::new()
    .with_config_modelpack_segdet(boxes_config, scores_config, seg_config)
    .build()?;

pub fn with_config_modelpack_segdet_split( self, boxes: Vec<Detection>, segmentation: Segmentation, ) -> Self

Loads a ModelPack segmentation split detection model configuration. Use DecoderBuilder.build() to parse the model configuration.

Examples

let config0 = configs::Detection {
    anchors: Some(vec![
        [0.36666667461395264, 0.31481480598449707],
        [0.38749998807907104, 0.4740740656852722],
        [0.5333333611488342, 0.644444465637207],
    ]),
    decoder: configs::DecoderType::ModelPack,
    quantization: Some(configs::QuantTuple(0.08547406643629074, 174)),
    shape: vec![1, 9, 15, 18],
    dshape: Vec::new(),
    normalized: Some(true),
};
let config1 = configs::Detection {
    anchors: Some(vec![
        [0.13750000298023224, 0.2074074000120163],
        [0.2541666626930237, 0.21481481194496155],
        [0.23125000298023224, 0.35185185074806213],
    ]),
    decoder: configs::DecoderType::ModelPack,
    quantization: Some(configs::QuantTuple(0.09929127991199493, 183)),
    shape: vec![1, 17, 30, 18],
    dshape: Vec::new(),
    normalized: Some(true),
};
let seg_config = configs::Segmentation {
    decoder: configs::DecoderType::ModelPack,
    quantization: Some(configs::QuantTuple(0.0064123, -31)),
    shape: vec![1, 640, 640, 2],
    dshape: Vec::new(),
};
let decoder = DecoderBuilder::new()
    .with_config_modelpack_segdet_split(vec![config0, config1], seg_config)
    .build()?;

pub fn with_config_modelpack_seg(self, segmentation: Segmentation) -> Self

Loads a ModelPack segmentation model configuration. Use DecoderBuilder.build() to parse the model configuration.

Examples

let seg_config = configs::Segmentation {
    decoder: configs::DecoderType::ModelPack,
    quantization: Some(configs::QuantTuple(0.0064123, -31)),
    shape: vec![1, 640, 640, 3],
    dshape: Vec::new(),
};
let decoder = DecoderBuilder::new()
    .with_config_modelpack_seg(seg_config)
    .build()?;

pub fn with_score_threshold(self, score_threshold: f32) -> Self

Sets the scores threshold of the decoder

Examples

let decoder = DecoderBuilder::new()
    .with_config_json_str(config_json)
    .with_score_threshold(0.654)
    .build()?;
assert_eq!(decoder.score_threshold, 0.654);

pub fn with_iou_threshold(self, iou_threshold: f32) -> Self

Sets the IOU threshold of the decoder. Has no effect when NMS is set to None

Examples

let decoder = DecoderBuilder::new()
    .with_config_json_str(config_json)
    .with_iou_threshold(0.654)
    .build()?;
assert_eq!(decoder.iou_threshold, 0.654);

pub fn with_nms(self, nms: Option<Nms>) -> Self

Sets the NMS mode for the decoder.

• Some(Nms::ClassAgnostic) — class-agnostic NMS (default): suppress overlapping boxes regardless of class label
• Some(Nms::ClassAware) — class-aware NMS: only suppress boxes that share the same class label AND overlap above the IoU threshold
• None — bypass NMS entirely (for end-to-end models with embedded NMS)

Examples

let decoder = DecoderBuilder::new()
    .with_config_json_str(config_json)
    .with_nms(Some(Nms::ClassAware))
    .build()?;
assert_eq!(decoder.nms, Some(Nms::ClassAware));

pub fn build(self) -> Result<Decoder, DecoderError>

Builds the decoder with the given settings. If the config is a JSON or YAML string, this will deserialize the JSON or YAML and then parse the configuration information.

Examples

let decoder = DecoderBuilder::new()
    .with_config_json_str(config_json)
    .with_score_threshold(0.654)
    .build()?;

Trait Implementations

impl Clone for DecoderBuilder

fn clone(&self) -> DecoderBuilder

Returns a duplicate of the value.

fn clone_from(&mut self, source: &Self)

Performs copy-assignment from source.

impl Debug for DecoderBuilder

fn fmt(&self, f: &mut Formatter<'_>) -> Result

Formats the value using the given formatter.

impl Default for DecoderBuilder

fn default() -> Self

Creates a default DecoderBuilder with no configuration and 0.5 score threshold and 0.5 OU threshold.

A valid confguration must be provided before building the Decoder.

Examples

let decoder = DecoderBuilder::default()
    .with_config_yaml_str(config_yaml)
    .build()?;
assert_eq!(decoder.score_threshold, 0.5);
assert_eq!(decoder.iou_threshold, 0.5);

impl PartialEq for DecoderBuilder

fn eq(&self, other: &DecoderBuilder) -> bool

Tests for self and other values to be equal, and is used by ==.

fn ne(&self, other: &Rhs) -> bool

Tests for !=. The default implementation is almost always sufficient, and should not be overridden without very good reason.

impl StructuralPartialEq for DecoderBuilder