Interface for running post-processing for Emotions.
Definition at line 142 of file Emotions.h.
Public Member Functions | |
EmotionsPostProcessor ()=delete | |
Removing the default constructor for EmotionsPostProcessor. More... | |
EmotionsPostProcessor (const ModelInputParams &modelInputParams, size_t numEmotions) | |
Constructor for EmotionsPostProcessor. More... | |
~EmotionsPostProcessor () | |
Destructor for EmotionsPostProcessor. More... | |
void | execute (Array< ArrayN< float, Emotions::TOP_EMOTIONS >> &emotionLikelihoods, Array< ArrayN< std::size_t, Emotions::TOP_EMOTIONS >> &topEmotions, const Tensor< CL, CX, F32 > &emotionsRaw, cudaStream_t stream=0) |
Running postprocessing for a given set of emotion likelihoods. More... | |
|
delete |
Removing the default constructor for EmotionsPostProcessor.
cvcore::emotions::EmotionsPostProcessor::EmotionsPostProcessor | ( | const ModelInputParams & | modelInputParams, |
size_t | numEmotions | ||
) |
Constructor for EmotionsPostProcessor.
modelInputParams | Model input parameters. |
numEmotions | number of the output emotions |
cvcore::emotions::EmotionsPostProcessor::~EmotionsPostProcessor | ( | ) |
Destructor for EmotionsPostProcessor.
void cvcore::emotions::EmotionsPostProcessor::execute | ( | Array< ArrayN< float, Emotions::TOP_EMOTIONS >> & | emotionLikelihoods, |
Array< ArrayN< std::size_t, Emotions::TOP_EMOTIONS >> & | topEmotions, | ||
const Tensor< CL, CX, F32 > & | emotionsRaw, | ||
cudaStream_t | stream = 0 |
||
) |
Running postprocessing for a given set of emotion likelihoods.
emotionLikelihoods | output emotions likelihood vector for each image in the batch. |
topEmotions | the top ranked emotions for each image in the batch |
emotionsRaw | input emotions vector in tensor format. |
stream | Cuda stream |