face_selector.h

#pragma once
 
#include <face_sdk_base/face.h>
#include <face_sdk_alignment/alignment.h>
 
namespace face_sdk
{
    enum class face_selection_algs
    {
        widest_face,            
        biggest_confidence,     
        closest_to_img_center   
    };
 
    class face_selector
    {
    public:
        virtual ~face_selector() = default;
 
        virtual std::shared_ptr<face_sdk::alignment_info> select_face(const face_sdk::alignment_info_set &alignment_infoes) const = 0;
 
        virtual std::shared_ptr<face_sdk::face> select_face(const face_sdk::face_set &faces) const = 0;
    };
 
    std::shared_ptr<face_selector> FACE_SDK_EXPORT make_face_selector(face_selection_algs alg, int32_t for_face_detector);
 
}