cnn_3dmm_expr.h

#ifndef FACE_SWAP_CNN_3DMM_EXPR_H
#define FACE_SWAP_CNN_3DMM_EXPR_H


#include "cnn_3dmm.h"

class FaceServices2;

namespace face_swap
{
    class CNN3DMMExpr : public CNN3DMM
    {
    public:

        CNN3DMMExpr(const std::string& deploy_file, const std::string& caffe_model_file,
            const std::string& mean_file, const std::string& model_file,
            bool generic = false, bool with_expr = true,
            bool with_gpu = true, int gpu_device_id = 0);

        ~CNN3DMMExpr();

        void process(const cv::Mat& img, const std::vector<cv::Point>& landmarks,
            cv::Mat& shape_coefficients, cv::Mat& tex_coefficients,
            cv::Mat& expr_coefficients, cv::Mat& vecR, cv::Mat& vecT, cv::Mat& K);
    private:
        std::unique_ptr<FaceServices2> fservice;
        bool m_generic, m_with_expr;
    };

}   // namespace face_swap

#endif // FACE_SWAP_CNN_3DMM_EXPR_H