cnn_3dmm.h

#ifndef FACE_SWAP_CNN_3DMM_H
#define FACE_SWAP_CNN_3DMM_H

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

// Caffe
#include <caffe/caffe.hpp>

#include <string>

namespace face_swap
{
    class CNN3DMM
    {
    public:

        CNN3DMM(const std::string& deploy_file, const std::string& caffe_model_file,
            const std::string& mean_file, bool init_cnn = true,
            bool with_gpu = true, int gpu_device_id = 0);

        ~CNN3DMM();

        void process(const cv::Mat& img, 
            cv::Mat& shape_coefficients, cv::Mat& tex_coefficients);

    private:
        void wrapInputLayer(std::vector<cv::Mat>& input_channels);

        cv::Mat readMean(const std::string& mean_file) const;

        cv::Mat preprocess(const cv::Mat& img);

        void copyInputData(cv::Mat& img);

    protected:
        std::shared_ptr<caffe::Net<float> > m_net;
        int m_num_channels;
        cv::Size m_input_size;
        cv::Mat m_mean;
        bool m_with_gpu;
    };

}   // namespace face_swap

#endif // FACE_SWAP_CNN_3DMM_H