HeadPose.h

Go to the documentation of this file.

/*
 * Copyright (c) 2021, NVIDIA CORPORATION & AFFILIATES. All rights reserved.
 *
 * NVIDIA CORPORATION, its affiliates and licensors retain all intellectual
 * property and proprietary rights in and to this material, related 
 * documentation and any modifications thereto. Any use, reproduction, 
 * disclosure or distribution of this material and related documentation
 * without an express license agreement from NVIDIA CORPORATION or 
 * its affiliates is strictly prohibited.
 * 
 * SPDX-FileCopyrightText: Copyright (c) 2021 NVIDIA CORPORATION & AFFILIATES
 * SPDX-License-Identifier: LicenseRef-NvidiaProprietary
 */
 
#ifndef CVCORE_HEADPOSE_H_
#define CVCORE_HEADPOSE_H_
 
#include <memory>
#include <vector>
 
#include <cv/core/Array.h>
#include <cv/core/CameraModel.h>
#include <cv/core/Core.h>
#include <cv/core/MathTypes.h>
 
namespace cvcore { namespace headpose {
 
struct HeadPoseParams
{
    size_t numLandmarks;                 
    std::vector<double> facePoints3d;    
    std::vector<int> landmarks2dIndices; 
};
 
class CVCORE_API HeadPose
{
public:
    HeadPose(const CameraIntrinsics &cameraParams);
 
    HeadPose(const HeadPoseParams &params, const CameraIntrinsics &cameraParams);
 
    ~HeadPose();
 
    void execute(Pose3d &outputPose, const Array<Vector2d> &inputLandmarks);
 
private:
    struct HeadPoseImpl;
    std::unique_ptr<HeadPoseImpl> m_pImpl;
};
 
}} // namespace cvcore::headpose
 
#endif