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 |