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 |