Model configuration
Note
These classes aren’t meant to be instantiated directly. They are used for module config validation with OmegaConf Structured config and provided here as a general reference of module config file format.
Base model configuration entities
Savant base models
Base deep learning model configuration templates.
![Inheritance diagram of Model, ObjectModel, AttributeModel, ComplexModel](../../_images/inheritance-d18d96a6bd9657f26963bb59a280f94cbda3f4c3.png)
Model child classes
![Inheritance diagram of ModelOutput, ObjectModelOutput, AttributeModelOutput, ComplexModelOutput](../../_images/inheritance-54811c231f51deee4527ddcb51660e13a2777266.png)
ModelOutput child classes
Base model configuration template. |
|
Model input parameters configuration template. |
|
Model output parameters configuration template. |
|
Object image preprocessing function configuration. |
|
Object model configuration template. |
|
ObjectModel output configuration template. |
|
ObjectModel output objects configuration template. |
|
Attribute model configuration template. |
|
AttributeModel output configuration template. |
|
AttributeModel output attribute configuration template. |
|
Complex model configuration template. |
|
ComplexModel output configuration template. |
Deepstream base models
Base configuration template for a nvinfer model. |
Deepstream model configuration
Gst-nvinfer model configuration templates.
![Inheritance diagram of savant.base.model.Model, NvInferModel, NvInferDetector, NvInferInstanceSegmentation, NvInferAttributeModel, NvInferComplexModel](../../_images/inheritance-99e2033c8479a3bd6ec7ed2370e26c06b86a78a4.png)
Model hierarchy with nvinfer entities
![Inheritance diagram of savant.base.model.ComplexModelOutput, NvInferObjectModelOutput](../../_images/inheritance-922e9b5ca0f542feb70855d6f4a4bbe3aed05b79.png)
Model output hierarchy with nvinfer entities
Note
The following classes specify configuration templates for fully realized concrete model types that can be included in the pipeline.
For example, base model types Model
or NvInferModel
cannot be used in configuration, while their descendant NvInferDetector
can.
Standard detector with orthogonal bboxes configuration template. |
|
NvInferAttribute model configuration template. |
|
NvInferComplexModel configuration template. |
|
nvinfer model input configuration template. |
|
NvInferObjectModel output configuration template for detector with aligned bboxes. |
|
NvInferObjectModel output objects configuration template.. |
- class savant.deepstream.nvinfer.model.NvInferModelFormat(value)
Enum for format of the provided model file.
- CAFFE = 0
Caffe model.
- UFF = 1
UFF model.
- ONNX = 2
ONNX model.
- ETLT = 3
Nvidia TAO model.
- CUSTOM = 4
Other format.
- class savant.deepstream.nvinfer.model.NvInferModelType(value)
Enum for type of the model (
network-type
in nvinfer configuration).- DETECTOR = 0
Detector model.
- CLASSIFIER = 1
Classification model.
- SEGMENTATION = 2
Semantic segmentation model.
- INSTANCE_SEGMENTATION = 3
Instance segmentation model.
- CUSTOM = 100
Custom model.