utilities.h

Go to the documentation of this file.

#ifndef __SFL_UTILITIES__
#define __SFL_UTILITIES__

// sfl
#include "sequence_face_landmarks.h"

namespace sfl
{
    void render(cv::Mat& img, const std::vector<cv::Point>& landmarks,
        bool drawLabels = false, const cv::Scalar& color = cv::Scalar(0, 255, 0),
        int thickness = 1);

    void render(cv::Mat& img, const cv::Rect& bbox,
        const cv::Scalar& color = cv::Scalar(0, 255, 0), int thickness = 1);

    void render(cv::Mat& img, const Face& face, bool drawIDs = true,
        bool drawLabels = false, const cv::Scalar& bbox_color = cv::Scalar(0, 0, 255),
        const cv::Scalar& landmarks_color = cv::Scalar(0, 255, 0), int thickness = 1,
        double fontScale = 1.0);

    void render(cv::Mat& img, const Frame& frame, bool drawIDs = true,
        bool drawLabels = false, const cv::Scalar& bbox_color = cv::Scalar(0, 0, 255),
        const cv::Scalar& landmarks_color = cv::Scalar(0, 255, 0), int thickness = 1,
        double fontScale = 1.0);

    void renderFaceID(cv::Mat& img, const Face& face, const cv::Scalar& color, int thickness = 1,
        double fontScale = 1.0);


    struct FaceStat
    {
        int id = 0;                 

        float avg_center_dist = 0;  
        int frame_count = 0;        
        float avg_size = 0;         

        float central_ratio = 0;    
        float frame_ratio = 0;      
        float size_ratio = 0;       
    };

    void getSequenceStats(const std::list<std::unique_ptr<Frame>>& sequence,
        std::vector<FaceStat>& stats);

    int getMainFaceID(const std::list<std::unique_ptr<Frame>>& sequence);

    int getMainFaceID(const std::vector<FaceStat>& stats);

    cv::Point2f getFaceLeftEye(const std::vector<cv::Point>& landmarks);

    cv::Point2f getFaceRightEye(const std::vector<cv::Point>& landmarks);

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

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

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

    cv::Point3f getFaceApproxEulerAngles(const std::vector<cv::Point>& landmarks);

    cv::Rect getFaceBBoxFromLandmarks(const std::vector<cv::Point>& landmarks, 
        const cv::Size& frameSize, bool square);

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

}   // namespace sfl

#endif  // __SFL_UTILITIES__