face_seg.h

#ifndef FACE_SEG_FACE_SEG_H
#define FACE_SEG_FACE_SEG_H

// std
#include <string>

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

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

namespace face_seg
{
    class FaceSeg
    {
    public:
        FaceSeg(const std::string& deploy_file, const std::string& model_file,
            bool with_gpu = true, int gpu_device_id = 0);

        ~FaceSeg();

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

    private:

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

        void preprocess(const cv::Mat& img, std::vector<cv::Mat>& input_channels);

        void subtractMean_c3(cv::Mat& img);

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

        // Mean pixel color
        const float MB = 104.00699f, MG = 116.66877f, MR = 122.67892f;
    };

}   // namespace face_seg

#endif // FACE_SEG_FACE_SEG_H