Defined in File add_classification.hpp
Base Type
public morpheus::AddScoresStageBase(Class AddScoresStageBase)
-
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
-
AddClassificationsStage(std::map<std::size_t, std::string> idx2label, float threshold)