quality_checker.h

#pragma once

#include <face_sdk_base/export.h>

#include <face_sdk_alignment/alignment.h>
#include <face_sdk_base/license.h>
#include <face_sdk_base/data_container.h>

#include <string>

namespace face_sdk
{
    struct quality_check_result
    {
        float hot_spots;        
        float blur;             
        float overexposure;     
        float rotation;         
        float glasses;          
        float smile;            
        float occlusion;        
        float closed_eyes;      
        float red_left_eye;     
        float red_right_eye;    
        float face_color;       
        float background;       
        float mask;             
    };

    typedef std::vector<quality_check_result> quality_check_result_set;

    class quality_checker
    {
    public:
        virtual ~quality_checker() = default;

        virtual quality_check_result_set check_faces(const face_set &faces) = 0;

        virtual bool is_compatible(uint32_t detector_version) const = 0;
    };

    struct quality_checker_initializer
    {
        bool use_blur_checker = true;                           
        bool use_face_color_checker = true;                     
        bool use_glasses_smile_occlusion_checker = true;        
        bool use_hot_spots_checker = true;                      
        bool use_overexposure_checker = true;                   
        bool use_red_eyes_checker = true;                       
        bool use_rotation_checker = true;                       
        bool use_background_checker = true;                     
        bool use_mask_checker = true;                           

        std::shared_ptr<face_sdk::data_container> face_color_container = nullptr;               
        std::shared_ptr<face_sdk::data_container> glasses_smile_occlusion_container = nullptr;  
        std::shared_ptr<face_sdk::data_container> red_eyes_container = nullptr;                 
        std::shared_ptr<face_sdk::data_container> background_container = nullptr;               
        std::shared_ptr<face_sdk::data_container> mask_container = nullptr;                     
    };

    std::shared_ptr<quality_checker> FACE_SDK_EXPORT make_quality_checker(
        uint32_t version,
        const std::shared_ptr<face_sdk::license> &license,
        const quality_checker_initializer &initializer,
        uint32_t batch_size,
        int32_t compute_device);

    std::shared_ptr<quality_checker> FACE_SDK_EXPORT make_quality_checker(
        uint32_t version,
        const quality_checker_initializer &initializer,
        uint32_t batch_size,
        int32_t compute_device);

    std::shared_ptr<quality_checker> FACE_SDK_EXPORT make_quality_checker(
        uint32_t version,
        const std::shared_ptr<face_sdk::license> &license,
        uint32_t batch_size,
        int32_t compute_device);
    

    std::shared_ptr<quality_checker> FACE_SDK_EXPORT make_quality_checker(
        uint32_t version,
        uint32_t batch_size,
        int32_t compute_device);

}