basel_3dmm.h

#ifndef FACE_SWAP_BASEL_3DMM_H
#define FACE_SWAP_BASEL_3DMM_H

// Includes
#include <opencv2/core.hpp>

#include <string>

namespace face_swap
{
    struct Mesh
    {
        static void save_ply(const Mesh& mesh, const std::string& ply_file);

        cv::Mat vertices;
        cv::Mat colors;
        cv::Mat faces;
        cv::Mat uv;
        cv::Mat tex;
        cv::Mat normals;
    };

    struct Basel3DMM
    {
        Mesh sample(const cv::Mat& shape_coefficients,
            const cv::Mat& tex_coefficients);

        Mesh sample(const cv::Mat& shape_coefficients,
            const cv::Mat& tex_coefficients, const cv::Mat& expr_coefficients);

        static Basel3DMM load(const std::string& model_file);

        cv::Mat faces;
        cv::Mat shapeMU, shapePC, shapeEV;
        cv::Mat texMU, texPC, texEV;
        cv::Mat exprMU, exprPC, exprEV;
    };

}   // namespace face_swap

#endif // FACE_SWAP_BASEL_3DMM_H