FaceSDKManager.h 11 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362363364365366367368369370371372373374375376377378379380381382383384385386387388389390391392393394395396397398399400401402403404405406407408409410411412413414415416417418419420421422423424425426427428429430431432433434435436437438439440441442443444445446447448449450451452
  1. //
  2. // FaceSDKManager.h
  3. // IDLFaceSDK
  4. //
  5. // Created by Tong,Shasha on 2017/5/15.
  6. // Copyright © 2017年 Baidu. All rights reserved.
  7. //
  8. #import <Foundation/Foundation.h>
  9. #import <UIKit/UIKit.h>
  10. typedef NS_ENUM(NSInteger, FaceLivenessActionType) {
  11. FaceLivenessActionTypeLiveEye = 0,
  12. FaceLivenessActionTypeLiveMouth = 1,
  13. FaceLivenessActionTypeLiveYawRight = 2,
  14. FaceLivenessActionTypeLiveYawLeft = 3,
  15. FaceLivenessActionTypeLivePitchUp = 4,
  16. FaceLivenessActionTypeLivePitchDown = 5,
  17. FaceLivenessActionTypeLiveYaw = 6,
  18. FaceLivenessActionTypeNoAction = 7,
  19. };
  20. typedef NS_ENUM(NSUInteger, ResultCode) {
  21. ResultCodeOK,
  22. ResultCodePitchOutofDownRange, //头部偏低
  23. ResultCodePitchOutofUpRange, //头部偏高
  24. ResultCodeYawOutofLeftRange, //头部偏左
  25. ResultCodeYawOutofRightRange, //头部偏右
  26. ResultCodeTooBrightIllumination, // 光线过亮
  27. ResultCodePoorIllumination, //光照不足
  28. ResultCodeNoFaceDetected, //没有检测到人脸
  29. ResultCodeDataNotReady,
  30. ResultCodeDataHitOne, //采集到一张照片
  31. ResultCodeDataHitLast, //采集到最后一张照片
  32. ResultCodeImageBlured, //图像模糊
  33. ResultCodeOcclusionLeftEye, //左眼有遮挡
  34. ResultCodeOcclusionRightEye, //右眼有遮挡
  35. ResultCodeOcclusionNose, //鼻子有遮挡
  36. ResultCodeOcclusionMouth, //嘴巴有遮挡
  37. ResultCodeOcclusionLeftContour, //左脸颊有遮挡
  38. ResultCodeOcclusionRightContour, //右脸颊有遮挡
  39. ResultCodeOcclusionChinCoutour, //下颚有遮挡
  40. ResultCodeVerifyInitError, //鉴权失败
  41. ResultCodeVerifyDecryptError,
  42. ResultCodeVerifyInfoFormatError,
  43. ResultCodeVerifyExpired,
  44. ResultCodeVerifyMissRequiredInfo,
  45. ResultCodeVerifyInfoCheckError,
  46. ResultCodeVerifyLocalFileError,
  47. ResultCodeVerifyRemoteDataError,
  48. ResultCodeLeftEyeClosed,
  49. ResultCodeRightEyeClosed,
  50. ResultCodeUnknowType //未知类型
  51. };
  52. typedef NS_ENUM(NSUInteger, TrackResultCode) {
  53. TrackResultCodeOK,
  54. TrackResultCodeImageBlured, // 图像模糊
  55. TrackResultCodePoorIllumination, // 光照不行
  56. TrackResultCodeNoFaceDetected, //没有检测到人脸
  57. TrackResultCodeOcclusionLeftEye, //左眼有遮挡
  58. TrackResultCodeOcclusionRightEye, //右眼有遮挡
  59. TrackResultCodeOcclusionNose, //鼻子有遮挡
  60. TrackResultCodeOcclusionMouth, //嘴巴有遮挡
  61. TrackResultCodeOcclusionLeftContour, //左脸颊有遮挡
  62. TrackResultCodeOcclusionRightContour, //右脸颊有遮挡
  63. TrackResultCodeOcclusionChinCoutour, //下颚有遮挡
  64. TrackResultCodeVerifyInitError, //鉴权失败
  65. TrackResultCodeVerifyDecryptError,
  66. TrackResultCodeVerifyInfoFormatError,
  67. TrackResultCodeVerifyExpired,
  68. TrackResultCodeVerifyMissRequiredInfo,
  69. TrackResultCodeVerifyInfoCheckError,
  70. TrackResultCodeVerifyLocalFileError,
  71. TrackResultCodeVerifyRemoteDataError,
  72. TrackResultCodeUnknowType //未知类型
  73. };
  74. @class FaceInfo;
  75. @class FaceLivenessState;
  76. @class FaceCropImageInfo;
  77. @interface FaceSDKManager : NSObject
  78. /* 超时时间 */
  79. @property (nonatomic, assign) CGFloat conditionTimeout;
  80. /* 语音超时*/
  81. @property (nonatomic, assign) CGFloat intervalOfVoiceRemind;
  82. /* 输出图像个数 */
  83. @property (nonatomic, assign) int imageNum;
  84. /* 图像加密类型,默认0 */
  85. @property (nonatomic, assign) int imageEncrypteType;
  86. /* 图像加密类型,默认0 */
  87. @property (nonatomic, assign) float minRectScale;
  88. /* 图片计数器,用来计当前图片的数量*/
  89. @property (nonatomic, assign) int currentNum;
  90. + (instancetype)sharedInstance;
  91. /**
  92. * 获取版本号
  93. */
  94. - (NSString *)getVersion;
  95. /**
  96. * 重置计数器
  97. */
  98. - (void)reset;
  99. /**
  100. * 获取设备zid 公安验证上传
  101. */
  102. - (NSString *)getZtoken;
  103. /**
  104. * SDK鉴权方法-文件授权
  105. * SDK鉴权方法 必须在使用其他方法之前设置,否则会导致SDK不可用
  106. *
  107. * @param licenseID 授权ID
  108. * @param licensePath 本地鉴权文件路径
  109. * @param remoteAuthorize 是否远程更新过期鉴权文件
  110. */
  111. - (void)setLicenseID:(NSString *)licenseID andLocalLicenceFile:(NSString *)licensePath andRemoteAuthorize:(BOOL)remoteAuthorize;
  112. /**
  113. * 初始化采集功能
  114. */
  115. - (int) initCollect;
  116. /**
  117. * 卸载采集功能
  118. */
  119. - (int)uninitCollect;
  120. /**
  121. * 判断授权是否通过,true 表示通过,false 表示不通过
  122. */
  123. - (BOOL)canWork;
  124. /**
  125. * 设置预测库耗能模式
  126. * 默认 LITE_POWER_NO_BIND
  127. */
  128. - (void)setLitePower:(int)litePower;
  129. /**
  130. * 需要检测的最大人脸数目
  131. * 默认1
  132. */
  133. - (void)setMaxDetectNum:(int)detectNum ;
  134. /**
  135. * 需要检测的最小人脸大小
  136. * 默认40
  137. */
  138. - (void)setMinFaceSize:(int)width;
  139. /**
  140. * 人脸置信度阈值(检测分值大于该阈值认为是人脸
  141. * RGB
  142. * 默认 0.5f
  143. */
  144. - (void)setNotFaceThreshold:(CGFloat)thr ;
  145. /**
  146. * 质量检测遮挡阈值
  147. * 默认0.5
  148. */
  149. - (void)setOccluThreshold:(CGFloat)thr ;
  150. /**
  151. * 质量检测遮挡阈值-左眼遮挡置信度
  152. * 默认0.31
  153. */
  154. -(void)setOccluLeftEyeThreshold:(CGFloat)thr ;
  155. /**
  156. * 质量检测遮挡阈值- 右眼遮挡置信度
  157. * 默认0.31
  158. */
  159. -(void)setOccluRightEyeThreshold:(CGFloat)thr ;
  160. /**
  161. * 质量检测遮挡阈值-鼻子遮挡置信度
  162. * 默认0.27
  163. */
  164. -(void)setOccluNoseThreshold:(CGFloat)thr ;
  165. /**
  166. * 质量检测遮挡阈值-嘴巴遮挡置信度
  167. * 默认0.2
  168. */
  169. -(void)setOccluMouthThreshold:(CGFloat)thr ;
  170. /**
  171. * 质量检测遮挡阈值-左脸遮挡置信度
  172. * 默认0.48
  173. */
  174. -(void)setOccluLeftCheekThreshold:(CGFloat)thr ;
  175. /**
  176. * 质量检测遮挡阈值-右脸遮挡置信度
  177. * 默认0.48
  178. */
  179. -(void)setOccluRightCheekThreshold:(CGFloat)thr ;
  180. /**
  181. * 质量检测遮挡阈值-下巴遮挡置信度
  182. * 默认0.4
  183. */
  184. -(void)setOccluChinThreshold:(CGFloat)thr ;
  185. /**
  186. * 最大光照阈值
  187. */
  188. - (void)setMaxIllumThreshold:(CGFloat)thr;
  189. /**
  190. * 最小光照阈值
  191. */
  192. - (void)setMinIllumThreshold:(CGFloat)thr ;
  193. /**
  194. * 质量检测模糊阈值
  195. * 默认0.5
  196. */
  197. - (void)setBlurThreshold:(CGFloat)thr ;
  198. /**
  199. * 姿态检测阈值
  200. * 默认pitch=12,yaw=12,row=10
  201. */
  202. - (void)setEulurAngleThrPitch:(float)pitch yaw:(float)yaw roll:(float)roll ;
  203. /**
  204. * 输出图像个数
  205. * 默认3
  206. */
  207. - (void)setMaxCropImageNum:(int)imageNum ;
  208. /**
  209. * 输出图像宽,设置为有效值(大于0)则对图像进行缩放,否则输出原图抠图结果
  210. * 默认 480
  211. */
  212. - (void)setCropFaceSizeWidth:(CGFloat)width ;
  213. /**
  214. * 输出图像高,设置为有效值(大于0)则对图像进行缩放,否则输出原图抠图结果
  215. * 默认 680
  216. */
  217. - (void)setCropFaceSizeHeight:(CGFloat)height ;
  218. /**
  219. * 输出图像,下巴扩展,大于等于0,0:不进行扩展
  220. * 默认0.1
  221. */
  222. - (void)setCropChinExtend:(CGFloat)chinExtend ;
  223. /**
  224. * 输出图像,额头扩展,大于等于0,0:不进行扩展
  225. * 默认0.2
  226. */
  227. - (void)setCropForeheadExtend:(CGFloat)foreheadExtend ;
  228. /**
  229. * 输出图像,人脸框与背景比例,大于等于1,1:不进行扩展
  230. * 默认1.5f
  231. */
  232. - (void)setCropEnlargeRatio:(float)cropEnlargeRatio;
  233. /**
  234. * 动作超时配置
  235. */
  236. - (void)setConditionTimeout:(CGFloat)timeout ;
  237. /**
  238. * 语音间隔提醒配置
  239. */
  240. - (void)setIntervalOfVoiceRemind:(CGFloat)timeout;
  241. /**
  242. * 是否开启静默活体,默认false
  243. */
  244. // - (void)setIsCheckSilentLive:(BOOL)isCheck;
  245. /**
  246. * 静默活体阈值配置,默认0.8。
  247. * 大于阈值返回图片,低于阈值返回未检测到人脸
  248. */
  249. // - (void)setSilentLiveThreshold:(CGFloat)thr ;
  250. /**
  251. * 是否开启口罩检测,非动作活体检测模型true,动作活体检测模型false
  252. */
  253. - (void)setIsCheckMouthMask:(BOOL)isCheck;
  254. /**
  255. * 口罩检测阈值配置,默认0.8。
  256. * 大于阈值判定为戴口罩,低于阈值判定为未戴口罩
  257. */
  258. - (void)setMouthMaskThreshold:(CGFloat)thr ;
  259. /**
  260. * 设置原始图片缩放比例,默认1不缩放,scale 阈值0~1
  261. */
  262. - (void)setImageWithScale:(CGFloat)scale;
  263. /**
  264. * 设置图片加密类型,type=0 基于base64 加密;type=1 基于百度安全算法加密
  265. */
  266. - (void)setImageEncrypteWithType:(int) type;
  267. /**
  268. * 人脸过远框比例 默认:0.4
  269. */
  270. - (void)setMinRect:(float) minRectScale;
  271. /**
  272. * 采集动作验证
  273. * @param image 检测的图片
  274. * @param isOriginal 是否返回原始图片
  275. * @param completion 判断采集是否完成,人脸信息状态是否正常
  276. */
  277. - (void)detectWithImage:(UIImage *)image isRreturnOriginalValue:(BOOL) isOriginal completion:(void (^)(FaceInfo *faceinfo, ResultCode resultCode))completion;
  278. /**
  279. * 动作活体动作验证
  280. * @param image 检测的图片
  281. * @param actionLiveType 当前要求做的动作
  282. * @param completion 判断当前动作是否完成,人脸信息状态是否正常
  283. */
  284. - (void)livenessWithImage:(UIImage *)image withAction:(FaceLivenessActionType)actionLiveType completion:(void (^)(FaceInfo *faceinfo, FaceLivenessState *state, ResultCode resultCode))completion;
  285. @end
  286. @interface FaceLivenessState : NSObject
  287. /**
  288. * 动作活体-眨眨眼
  289. */
  290. @property(nonatomic, assign) BOOL isLiveEye;
  291. /**
  292. * 动作活体-张张嘴
  293. */
  294. @property(nonatomic, assign) BOOL isLiveMouth;
  295. /**
  296. * 动作活体-向左转头
  297. */
  298. @property(nonatomic, assign) BOOL isLiveYawLeft;
  299. /**
  300. * 动作活体-向右转头
  301. */
  302. @property(nonatomic, assign) BOOL isLiveYawRight;
  303. /**
  304. * 动作活体-抬头
  305. */
  306. @property(nonatomic, assign) BOOL isLivePitchUp;
  307. /**
  308. * 动作活体-低头
  309. */
  310. @property(nonatomic, assign) BOOL isLivePitchDown;
  311. @end
  312. @interface FaceCropImageInfo : NSObject
  313. /**
  314. * 基于质量检测,姿态角度,对图片得分
  315. */
  316. @property (nonatomic, assign) float qualityScore;
  317. /**
  318. * 离线RGB 静默活体得分
  319. */
  320. @property (nonatomic,assign) float silentliveScore;
  321. /**
  322. * 采集到的矫正,调整宽高图片,会有宽高
  323. */
  324. @property (nonatomic ,strong) UIImage *cropImageWithBlack;
  325. /**
  326. * 加密采集到的矫正,调整宽高图片,会有宽高
  327. */
  328. @property (nonatomic ,strong) NSString *cropImageWithBlackEncryptStr;
  329. /**
  330. * 原始图片
  331. */
  332. @property (nonatomic ,strong) UIImage *originalImage;
  333. /**
  334. * 加密原始图片
  335. */
  336. @property (nonatomic ,strong) NSString *originalImageEncryptStr;
  337. /**
  338. * 排序规则
  339. */
  340. - (NSComparisonResult)compareWithImageInfo:(FaceCropImageInfo *)info;
  341. @end
  342. @interface FaceInfo : NSObject
  343. /**
  344. * 人脸在图片中的位置
  345. */
  346. @property (nonatomic, assign) CGRect faceRect;
  347. /**
  348. * 人脸track 中的faceid
  349. */
  350. @property (nonatomic, assign) NSInteger faceId;
  351. /**
  352. * 人脸72关键点
  353. */
  354. @property (nonatomic, strong) NSArray * landMarks;
  355. /**
  356. * 人脸角度
  357. */
  358. @property (nonatomic, assign) float angle;
  359. /**
  360. * 人脸质量-光照置信度,通过quality 方法调用获取
  361. */
  362. @property (nonatomic,assign) float illum;
  363. /**
  364. * 人脸质量-模糊置信度,通过quality 方法调用获取
  365. */
  366. @property (nonatomic,assign) float blur;
  367. /**
  368. * 人脸上下偏转角,通过headpose 方法调用获取
  369. */
  370. @property (nonatomic, assign) float pitch;
  371. /**
  372. * 人脸左右偏转角,通过headpose 方法调用获取
  373. */
  374. @property (nonatomic, assign) float yaw;
  375. /**
  376. * 人脸平行平面内的头部旋转角,通过headpose 方法调用获取
  377. */
  378. @property (nonatomic, assign) float roll;
  379. /**
  380. * 离线RGB 静默活体得分
  381. */
  382. @property (nonatomic,assign) float silentliveScore;
  383. /**
  384. * 人脸检测得分
  385. */
  386. @property (nonatomic, assign) CGFloat score;
  387. /**
  388. * 输出图片结构体,包含图片质量分数,裁剪没有黑边的图片,裁剪有黑边的图片,未裁剪图片,原始图
  389. */
  390. @property (nonatomic, strong) FaceCropImageInfo * cropImageInfo;
  391. @end