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 algorithm, uint32_t face_detector_version);

}