utilities.h

Go to the documentation of this file.

#ifndef __FVG_UTILITIES__
#define __FVG_UTILITIES__

#include <segment_util/segmentation_util.h>
#include <face_video_segment.pb.h>

namespace fvs
{
    void createContours(const segmentation::VectorMesh& mesh,
        const segmentation::SegmentationDesc_Polygon& poly,
        std::vector<std::vector<cv::Point>>& contours);

    cv::Mat calcSegmentation(cv::Size size,
        const google::protobuf::Map<google::protobuf::uint32, Region>& regions,
        const segmentation::SegmentationDesc& seg_desc);

    cv::Mat calcSegmentation(const cv::Mat& face_map,
        const google::protobuf::Map<google::protobuf::uint32, Region>& regions,
        const segmentation::SegmentationDesc& seg_desc);

    void postprocessSegmentation(cv::Mat& seg, bool disconnected = true,
        bool holes = true, bool smooth = true, int smooth_iterations = 1,
        int smooth_kernel_radius = 2);

//    void createFullFace(const std::vector<cv::Point>& landmarks,
//        std::vector<cv::Point>& full_face);

    void renderBoundaries(cv::Mat& img, int hierarchy_level,
        const segmentation::SegmentationDesc& desc,
        const segmentation::Hierarchy* seg_hier,
        const cv::Scalar& color = cv::Scalar(255, 255, 255));

    void renderSegmentation(cv::Mat& img, const cv::Mat& seg, uchar color);

    void renderSegmentationBlend(cv::Mat& img, const cv::Mat& seg, float alpha = 0.5f,
        const cv::Scalar& full_color = cv::Scalar(0, 0, 255),
        const cv::Scalar& intersection_color = cv::Scalar(255, 0, 0));

    float getFaceDominantSide(const std::vector<cv::Point>& landmarks);

    cv::Rect getFaceBBoxFromSegmentation(const cv::Mat& seg, bool square);

}   // namespace fvs

#endif  // __FVG_UTILITIES__