Class AddClassificationsStage

• Defined in File add_classification.hpp

Inheritance Relationships

Base Type

• public morpheus::AddScoresStageBase (Class AddScoresStageBase)

Class Documentation

class AddClassificationsStage : public morpheus::AddScoresStageBase

Add detected classifications to each message. Classification labels based on probabilities calculated in inference stage. Label indexes will be looked up in the idx2label property.

Public Functions

AddClassificationsStage(

std::map<std::size_t, std::string> idx2label,

float threshold

)

Construct a new Add Classifications Stage object.

Parameters:

• threshold – : Threshold to consider true/false for each class

• idx2label – : Index to classification labels map