100 lines
2.5 KiB
C++
100 lines
2.5 KiB
C++
//
|
||
// Created by sunnydu on 2024/5/9.
|
||
//
|
||
|
||
#ifndef VERIFICATION_YTFACEQUALITYDETECT_H
|
||
#define VERIFICATION_YTFACEQUALITYDETECT_H
|
||
#include "DetectFaceInfo.h"
|
||
#ifdef __ANDROID__
|
||
#include <core.hpp>
|
||
#include <resize.hpp>
|
||
#include "cvtColor.hpp"
|
||
|
||
#else
|
||
#include <YTCv/core.hpp>
|
||
#include <YTCv/cvtColor.hpp>
|
||
#include <YTCv/resize.hpp>
|
||
#endif // __ANDROID__
|
||
#include <vector>
|
||
class YTFaceQualityDetect {
|
||
public:
|
||
int needFaceQualityImage = false;
|
||
float stableRoiThreshold = 0.98f;//模糊判断
|
||
float continuousShelterNumThreshold = 5;
|
||
int continuousQualityNumThreshold = 10;
|
||
// float faceMaxHeightThreshold = 0.95f;
|
||
// float faceRealMinHeightThreshold = 0.4f;
|
||
float secondaryYawThreshold = 18;
|
||
float secondaryPitchThreshold = 18;
|
||
float secondaryRollThreshold = 18;
|
||
float closeMouthThreshold = 0.4f;
|
||
float closeEyeLeftThreshold = 0.25f;//左睁眼阈值
|
||
float closeEyeRightThreshold = 0.25f;//右睁眼阈值
|
||
//人脸质量最佳帧
|
||
bool isGetImageMat = false;
|
||
yt_tinycv::Mat3BGR qualityImageMat;
|
||
std::vector<float> qualityShape;
|
||
int continuousCount = 0;
|
||
int continuousShelterCount = 0;
|
||
/**
|
||
* 清空连续帧计数
|
||
*/
|
||
void resetContinuousCount();
|
||
/**
|
||
* 遮挡累计检测
|
||
*/
|
||
void shelterDetect();
|
||
/**
|
||
* 计数器+1，然后判断是否满足条件
|
||
*/
|
||
int faceQualityIsPass(DetectFaceInfo& detectFaceInfo);
|
||
private:
|
||
yt_tinycv::Rect2i previousFaceRect = yt_tinycv::Rect2i(0,0,0,0);
|
||
/**
|
||
* 内置二级角度检测
|
||
*/
|
||
bool faceAngleForceCheck(DetectFaceInfo& detectFaceInfo);
|
||
/**
|
||
* 检测脸部长度占比，脸部长度占比在图像长度的预设百分比(faceHeightThreshold) 视为通过，否则不通过
|
||
* faceHeightThreshold 默认为0，不对人脸大小做限制
|
||
*
|
||
* @return 1 太远
|
||
* 2 太近
|
||
* 0 合适
|
||
*/
|
||
int isFaceHeightStandard(DetectFaceInfo& detectFaceInfo);
|
||
|
||
/**
|
||
* 闭眼判断
|
||
* @param shape
|
||
* @return
|
||
*/
|
||
bool isEyeOpen(DetectFaceInfo& detectFaceInfo);
|
||
|
||
/**
|
||
* 张嘴检测
|
||
* @param detectFaceInfo
|
||
* @return
|
||
*/
|
||
bool isMouthClose(DetectFaceInfo& detectFaceInfo);
|
||
|
||
/**
|
||
* 计算人脸框差值
|
||
* @param rect1
|
||
* @param outRect
|
||
*/
|
||
void getIntersectionRect(const yt_tinycv::Rect2i& rect1,yt_tinycv::Rect2i& outRect);
|
||
|
||
/**
|
||
* 遮挡判断
|
||
* @param rect
|
||
* @return
|
||
*/
|
||
bool detectScreenShaking(const yt_tinycv::Rect2i& rect);
|
||
|
||
|
||
};
|
||
|
||
|
||
#endif //VERIFICATION_YTFACEQUALITYDETECT_H
|