Lens AI Profiler Cpp
Public Member Functions | Public Attributes | List of all members
ModelProfile Class Reference

Class for computing and managing various model statistics. More...

Public Member Functions

 ModelProfile (std::string model_id, std::string conf_path, int save_interval, int top_classes)
 Constructor to initialize ModelProfile object. More...
 
int log_classification_model_stats (float inference_latency, const ClassificationResults &results)
 Logs statistics for a classification model. More...
 
int getNumDistributionBoxes () const
 
const distributionBoxgetDistributionBox (unsigned int index) const
 

Public Attributes

frequent_class_sketchsketch1
 Logs statistics for a YOLOv5 model. More...
 

Detailed Description

Class for computing and managing various model statistics.

Constructor & Destructor Documentation

◆ ModelProfile()

ModelProfile::ModelProfile ( std::string  model_id,
std::string  conf_path,
int  save_interval,
int  top_classes 
)

Constructor to initialize ModelProfile object.

ModelProfile class constructor.

Parameters
model_idUnique identifier for the model
no_of_classesNumber of classes the model predicts
saverReference to a Saver object used for saving model statistics

Member Function Documentation

◆ log_classification_model_stats()

int ModelProfile::log_classification_model_stats ( float  inference_latency,
const ClassificationResults &  results 
)

Logs statistics for a classification model.

Logs classification model statistics.

Parameters
inference_latencyTime taken for model inference
resultsReference to the classification results
Returns
0 on success, negative value on error
Parameters
inference_latencyTime taken for model inference
resultsReference to the classification results
Returns
0 on success, negative value on error

This function iterates through the provided results and logs statistics for the most frequent classes. It updates the model_classes_stat map with scores for each class.

Member Data Documentation

◆ sketch1

frequent_class_sketch* ModelProfile::sketch1

Logs statistics for a YOLOv5 model.

Parameters
inference_latencyTime taken for model inference
resultsReference to the detection results from YOLOv5 model
Returns
0 on success, negative value on error

The documentation for this class was generated from the following files: