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
|