using System;
|
using System.Collections;
|
using System.Collections.Generic;
|
using System.IO;
|
using System.Linq;
|
using System.Runtime.InteropServices;
|
using System.Text;
|
using System.Threading.Tasks;
|
|
namespace WIDESEAWCS_Common.Face
|
{
|
public class FaceHelper
|
{
|
static string picturePath = "";
|
static FileInfo[] fileInfos;
|
static string currentPath = "";
|
|
/// <summary>
|
/// 人脸识别方法
|
/// </summary>
|
/// <param name="sourcePath">已录入人脸图片路径</param>
|
/// <param name="desPath">要是被的人脸图片路径</param>
|
/// <returns></returns>
|
public static (bool, string) FaceRecognition(string sourcePath, string desPath)
|
{
|
int ret = TH_Faces.IdFaceSdkInit();
|
if (ret < 0)
|
{
|
IntPtr pRunCode = Marshal.AllocHGlobal(35 + 1);
|
TH_Faces.IdFaceSdkGetRunCode(pRunCode);
|
Marshal.FreeHGlobal(pRunCode);
|
}
|
|
picturePath = sourcePath;
|
currentPath = desPath;
|
List<IntPtr> pFrame_Sources = new List<IntPtr>();
|
List<IntPtr> pFeature_Sources = new List<IntPtr>();
|
GetSourceFeature(ref pFrame_Sources, ref pFeature_Sources);
|
|
IntPtr pFrame_Desc = IntPtr.Zero;
|
IntPtr pFeature_Desc = IntPtr.Zero;
|
GetDesFeature(ref pFrame_Desc, ref pFeature_Desc);
|
|
List<byte> bytes = Check(ref pFrame_Desc, ref pFeature_Desc, ref pFrame_Sources, ref pFeature_Sources);
|
|
if (bytes.Count == 0)
|
{
|
return (false, "人脸识别失败");
|
}
|
|
byte max = bytes.Max(); // 获取最大分数
|
|
int index = bytes.IndexOf(max); // 获取最大分数的索引
|
|
if (fileInfos.Length > index)
|
{
|
if (max < 80) // 如果最大分数小于80，认为人脸识别失败
|
{
|
return (false, "人脸识别失败");
|
}
|
|
return (true, fileInfos[index].Name); // 返回对应的文件名
|
}
|
|
return (false, "人脸识别失败");
|
}
|
|
public static (bool, string) FaceRecognitionOne(string sourcePath, string desPath)
|
{
|
int ret = TH_Faces.IdFaceSdkInit();
|
if (ret < 0)
|
{
|
IntPtr pRunCode = Marshal.AllocHGlobal(35 + 1);
|
TH_Faces.IdFaceSdkGetRunCode(pRunCode);
|
Marshal.FreeHGlobal(pRunCode);
|
}
|
|
picturePath = sourcePath;
|
currentPath = desPath;
|
IntPtr pFrame_Source = IntPtr.Zero;
|
IntPtr pFeature_Source = IntPtr.Zero;
|
GetSourceFeature(ref pFrame_Source, ref pFeature_Source);
|
|
IntPtr pFrame_Desc = IntPtr.Zero;
|
IntPtr pFeature_Desc = IntPtr.Zero;
|
GetDesFeature(ref pFrame_Desc, ref pFeature_Desc);
|
|
byte result = Check(ref pFrame_Desc, ref pFeature_Desc, ref pFrame_Source, ref pFeature_Source);
|
|
|
if (result < 80) // 如果最大分数小于80，认为人脸识别失败
|
{
|
return (false, "人脸识别失败");
|
}
|
|
return (true, "人脸识别成功"); // 返回对应的文件名
|
}
|
|
private static void GetSourceFeature(ref List<IntPtr> pFrame_Sources, ref List<IntPtr> pFeature_Sources)
|
{
|
DirectoryInfo dir = new DirectoryInfo(picturePath);
|
fileInfos = dir.GetFiles();
|
|
foreach (var file in fileInfos)
|
{
|
// 读原始图象
|
int nWidth1 = 0;
|
int nHeight1 = 0;
|
IntPtr pSrcFilepath = Marshal.StringToHGlobalAnsi(file.FullName);
|
// 读图象分辨率
|
int ret = TH_Faces.ReadImageFile(pSrcFilepath, IntPtr.Zero, 0, ref nWidth1, ref nHeight1, 24);
|
if (ret < 0 || nWidth1 <= 0 || nHeight1 <= 0)
|
{
|
continue;
|
}
|
// 根据图象分辨率分配图象数据缓冲区并读出图象数据
|
IntPtr pFrame_Source = Marshal.AllocHGlobal(nWidth1 * nHeight1 * 3);
|
TH_Faces.ReadImageFile(pSrcFilepath, pFrame_Source, nWidth1 * nHeight1 * 3, ref nWidth1, ref nHeight1, 24);
|
if (ret < 0 || nWidth1 <= 0 || nHeight1 <= 0)
|
{
|
Marshal.FreeHGlobal(pFrame_Source);
|
continue;
|
}
|
|
// 对原始图象进行人脸检测
|
TH_Faces.FACE_DETECT_RESULT Face1 = new TH_Faces.FACE_DETECT_RESULT();
|
int nNum1 = TH_Faces.IdFaceSdkDetectFace(pFrame_Source, nWidth1, nHeight1, ref Face1);
|
if (nNum1 <= 0)
|
{
|
Marshal.FreeHGlobal(pFrame_Source);
|
continue;
|
}
|
|
// 对原始图象提取人脸特征（输入原始图象及其人脸检测结果）
|
int nFeaSize = TH_Faces.IdFaceSdkFeatureSize();
|
IntPtr pFeature_Source = Marshal.AllocHGlobal(nFeaSize);
|
ret = TH_Faces.IdFaceSdkFeatureGet(pFrame_Source, nWidth1, nHeight1, ref Face1, pFeature_Source);
|
if (ret != 0)
|
{
|
Marshal.FreeHGlobal(pFrame_Source);
|
Marshal.FreeHGlobal(pFeature_Source);
|
continue;
|
}
|
|
pFrame_Sources.Add(pFrame_Source);
|
pFeature_Sources.Add(pFeature_Source);
|
}
|
}
|
|
private static void GetSourceFeature(ref IntPtr pFrame_Source, ref IntPtr pFeature_Source)
|
{
|
// 读原始图象
|
int nWidth1 = 0;
|
int nHeight1 = 0;
|
IntPtr pSrcFilepath = Marshal.StringToHGlobalAnsi(picturePath);
|
// 读图象分辨率
|
int ret = TH_Faces.ReadImageFile(pSrcFilepath, IntPtr.Zero, 0, ref nWidth1, ref nHeight1, 24);
|
if (ret < 0 || nWidth1 <= 0 || nHeight1 <= 0)
|
{
|
return;
|
}
|
// 根据图象分辨率分配图象数据缓冲区并读出图象数据
|
pFrame_Source = Marshal.AllocHGlobal(nWidth1 * nHeight1 * 3);
|
TH_Faces.ReadImageFile(pSrcFilepath, pFrame_Source, nWidth1 * nHeight1 * 3, ref nWidth1, ref nHeight1, 24);
|
if (ret < 0 || nWidth1 <= 0 || nHeight1 <= 0)
|
{
|
Marshal.FreeHGlobal(pFrame_Source);
|
}
|
|
// 对原始图象进行人脸检测
|
TH_Faces.FACE_DETECT_RESULT Face1 = new TH_Faces.FACE_DETECT_RESULT();
|
int nNum1 = TH_Faces.IdFaceSdkDetectFace(pFrame_Source, nWidth1, nHeight1, ref Face1);
|
if (nNum1 <= 0)
|
{
|
Marshal.FreeHGlobal(pFrame_Source);
|
}
|
|
// 对原始图象提取人脸特征（输入原始图象及其人脸检测结果）
|
int nFeaSize = TH_Faces.IdFaceSdkFeatureSize();
|
pFeature_Source = Marshal.AllocHGlobal(nFeaSize);
|
ret = TH_Faces.IdFaceSdkFeatureGet(pFrame_Source, nWidth1, nHeight1, ref Face1, pFeature_Source);
|
if (ret != 0)
|
{
|
Marshal.FreeHGlobal(pFrame_Source);
|
Marshal.FreeHGlobal(pFeature_Source);
|
}
|
}
|
|
private static bool GetDesFeature(ref IntPtr pFrame_Desc, ref IntPtr pFeature_Desc)
|
{
|
// 读目标图象
|
int nWidth2 = 0;
|
int nHeight2 = 0;
|
IntPtr pDstFilepath = Marshal.StringToHGlobalAnsi(currentPath);
|
// 读图象分辨率
|
int ret = TH_Faces.ReadImageFile(pDstFilepath, IntPtr.Zero, 0, ref nWidth2, ref nHeight2, 24);
|
if (ret < 0 || nWidth2 <= 0 || nHeight2 <= 0)
|
{
|
return false;
|
}
|
// 根据图象分辨率分配图象数据缓冲区并读出图象数据
|
pFrame_Desc = Marshal.AllocHGlobal(nWidth2 * nHeight2 * 3);
|
ret = TH_Faces.ReadImageFile(pDstFilepath, pFrame_Desc, nWidth2 * nHeight2 * 3, ref nWidth2, ref nHeight2, 24);
|
if (ret < 0 || nWidth2 <= 0 || nHeight2 <= 0)
|
{
|
Marshal.FreeHGlobal(pFrame_Desc);
|
return false;
|
}
|
|
// 对目标图象进行人脸检测
|
TH_Faces.FACE_DETECT_RESULT Face2 = new TH_Faces.FACE_DETECT_RESULT();
|
int nNum2 = TH_Faces.IdFaceSdkDetectFace(pFrame_Desc, nWidth2, nHeight2, ref Face2);
|
if (nNum2 <= 0)
|
{
|
Marshal.FreeHGlobal(pFrame_Desc);
|
return false;
|
}
|
int nFeaSize = TH_Faces.IdFaceSdkFeatureSize();
|
pFeature_Desc = Marshal.AllocHGlobal(nFeaSize);
|
ret = TH_Faces.IdFaceSdkFeatureGet(pFrame_Desc, nWidth2, nHeight2, ref Face2, pFeature_Desc);
|
if (ret != 0)
|
{
|
Marshal.FreeHGlobal(pFrame_Desc);
|
Marshal.FreeHGlobal(pFeature_Desc);
|
return false;
|
}
|
|
return true;
|
}
|
|
private static List<byte> Check(ref IntPtr pFrame_Desc, ref IntPtr pFeature_Desc, ref List<IntPtr> pFrame_Sources, ref List<IntPtr> pFeature_Sources)
|
{
|
IntPtr hList = TH_Faces.IdFaceSdkListCreate(1000); // 创建支持1000个特征的比对列表
|
try
|
{
|
List<byte> results = new List<byte>();
|
int i, n = 0;
|
|
|
for (i = 0; i < pFeature_Sources.Count; i++)
|
{
|
// 将两个人的特征码循环交叉插入比对列表中
|
int nPos = -1;
|
n = TH_Faces.IdFaceSdkListInsert(hList, ref nPos, pFeature_Sources[i], 1);
|
}
|
if (n != pFeature_Sources.Count)
|
{
|
|
return results; ;
|
}
|
else
|
{
|
IntPtr pnScores = Marshal.AllocHGlobal(pFeature_Sources.Count);
|
n = TH_Faces.IdFaceSdkListCompare(hList, pFeature_Desc, 0, pFeature_Sources.Count, pnScores); // 现场人与加入列表中的 N 个人进行一对多比对
|
if (n != pFeature_Sources.Count)
|
{
|
|
return results;
|
}
|
else
|
{
|
byte[] scores = new byte[pFeature_Sources.Count];
|
Marshal.Copy(pnScores, scores, 0, pFeature_Sources.Count);
|
results = scores.ToList(); // 将比对结果转换为 List<byte> 类型
|
|
}
|
Marshal.FreeHGlobal(pnScores);
|
}
|
|
return results;
|
}
|
finally
|
{
|
TH_Faces.IdFaceSdkListDestroy(hList); // 销毁比对列表
|
Marshal.FreeHGlobal(pFrame_Desc);
|
Marshal.FreeHGlobal(pFeature_Desc);
|
|
foreach (var pFrame in pFrame_Sources)
|
{
|
|
Marshal.FreeHGlobal(pFrame);
|
|
}
|
|
foreach (var pFeature in pFeature_Sources)
|
{
|
|
Marshal.FreeHGlobal(pFeature);
|
}
|
|
TH_Faces.IdFaceSdkUninit();
|
}
|
}
|
|
|
private static byte Check(ref IntPtr pFrame_Desc, ref IntPtr pFeature_Desc, ref IntPtr pFrame_Source, ref IntPtr pFeature_Source)
|
{
|
try
|
{
|
byte score = TH_Faces.IdFaceSdkFeatureCompare(pFeature_Source, pFeature_Desc);
|
|
return score;
|
}
|
finally
|
{
|
Marshal.FreeHGlobal(pFrame_Desc);
|
Marshal.FreeHGlobal(pFeature_Desc);
|
Marshal.FreeHGlobal(pFrame_Source);
|
Marshal.FreeHGlobal(pFeature_Source);
|
TH_Faces.IdFaceSdkUninit();
|
}
|
}
|
}
|
}
|
