Public Member Functions | |
| GazeNetPreProcessor ()=delete | |
| Default constructor is deleted. More... | |
| GazeNetPreProcessor (const ImagePreProcessingParams &preProcessorParams, const ModelInputParams &modelInputParams) | |
| Constructor of GazeNetPreProcessor. More... | |
| ~GazeNetPreProcessor () | |
| Destructor of GazeNetPreProcessor. More... | |
| void | setLandmarksMeanAndVariance (const Array< Vector2f > &landmarksMean, const Array< Vector2f > &landmarksStd) |
| Set the mean and standard deviation of the landmarks for the model. More... | |
| void | execute (Tensor< NHWC, C1, F32 > &outputFace, Tensor< NHWC, C1, F32 > &outputLeft, Tensor< NHWC, C1, F32 > &outputRight, Tensor< CL, CX, F32 > &outputFeatures, const Tensor< NHWC, C3, U8 > &inputImage, const Array< BBox > &inputBBox, const Array< ArrayN< Vector2f, GazeNetPreProcessor::NUM_LANDMARKS >> &inputLandmarks, FeatureType type=FeatureType::FACEGRID, cudaStream_t stream=0) |
| Main interface to run pre-processing. More... | |
Static Public Attributes | |
| static constexpr size_t | NUM_LANDMARKS = 68 |
| Number of landmarks needed for the model. More... | |
| static constexpr float | BBOX_FACE_SCALE = 1.3f |
| Scaling factor for face. More... | |
| static constexpr float | BBOX_EYE_SCALE = 1.8f |
| Scaling factor for eyes. More... | |
| static const float | DEFAULT_LANDMARKS_MEAN [] |
| Default landmarks mean values for the model. More... | |
| static const float | DEFAULT_LANDMARKS_STD [] |
| Default landmarks standard deviation values for the model. More... | |
|
delete |
Default constructor is deleted.
| cvcore::gazenet::GazeNetPreProcessor::GazeNetPreProcessor | ( | const ImagePreProcessingParams & | preProcessorParams, |
| const ModelInputParams & | modelInputParams | ||
| ) |
Constructor of GazeNetPreProcessor.
| preProcessorParams | image pre-processing parameters. |
| modelInputParams | model paramters for network. |
| cvcore::gazenet::GazeNetPreProcessor::~GazeNetPreProcessor | ( | ) |
Destructor of GazeNetPreProcessor.
| void cvcore::gazenet::GazeNetPreProcessor::execute | ( | Tensor< NHWC, C1, F32 > & | outputFace, |
| Tensor< NHWC, C1, F32 > & | outputLeft, | ||
| Tensor< NHWC, C1, F32 > & | outputRight, | ||
| Tensor< CL, CX, F32 > & | outputFeatures, | ||
| const Tensor< NHWC, C3, U8 > & | inputImage, | ||
| const Array< BBox > & | inputBBox, | ||
| const Array< ArrayN< Vector2f, GazeNetPreProcessor::NUM_LANDMARKS >> & | inputLandmarks, | ||
| FeatureType | type = FeatureType::FACEGRID, |
||
| cudaStream_t | stream = 0 |
||
| ) |
Main interface to run pre-processing.
| outputFace | output tensor for face region. |
| outputLeft | output tensor for left eye. |
| outputRight | output tensor for right eye. |
| outputFeatures | output normalized landmarks or facegrid. |
| inputImage | input image tensor. |
| inputBBox | input BBox of the face. |
| inputLandmarks | input raw facial landmarks. |
| type | whether to use facegrid model or landmarks model. |
| stream | cuda stream. |
| void cvcore::gazenet::GazeNetPreProcessor::setLandmarksMeanAndVariance | ( | const Array< Vector2f > & | landmarksMean, |
| const Array< Vector2f > & | landmarksStd | ||
| ) |
Set the mean and standard deviation of the landmarks for the model.
| landmarksMean | mean of the landmarks. |
| landmarksStd | standard deviation of the landmarks. |
|
staticconstexpr |
|
staticconstexpr |
|
static |
|
static |
|
staticconstexpr |