using OpenCvSharp;
|
using System;
|
using System.Runtime.InteropServices;
|
/**
|
* 备注（人脸数据库管理说明）：
|
* 人脸数据库为采用sqlite3的数据库，会自动生成一个db目录夹，下面有数据库face.db文件保存数据库
|
* 可用sqliteExpert之类的可视化工具打开查看,亦可用命令行，方法请自行百度。
|
* 该数据库仅仅可适应于5w人左右的人脸库，且设计表格等属于小型通用化。
|
* 若不能满足客户个性化需求，客户可自行设计数据库保存数据。宗旨就是每个人脸图片提取一个特征值保存。
|
* 人脸1:1,1:N比对及识别实际就是特征值的比对。1:1只要提取2张不同的图片特征值调用compare_feature比对。
|
* 1：N是提取一个特征值和数据库中已保存的N个特征值一一比对(比对速度很快，不用担心效率问题)，
|
* 最终取分数高的值为最高相似度。
|
* 相似度识别的分数可自行测试根据实验结果拟定，一般推荐相似度大于80分为同一人。
|
*
|
*/
|
namespace FaceAI
|
{
|
class FaceManager
|
{
|
// 人脸注册(传特征值,特征值可参考FaceFeature.cs提取，亦可参考FaceCompare.cs查看特征值的比对)
|
[DllImport("BaiduFaceApi.dll", EntryPoint = "user_add", CharSet = CharSet.Ansi
|
, CallingConvention = CallingConvention.Cdecl)]
|
public static extern IntPtr user_add(ref BDFaceFeature f1, string user_id, string group_id,
|
string user_info="");
|
|
// 人脸注册(传opencv图片帧,特征值可参考FaceFeature.cs提取，亦可参考FaceCompare.cs查看特征值的比对)
|
[DllImport("BaiduFaceApi.dll", EntryPoint = "user_add_by_mat", CharSet = CharSet.Ansi
|
, CallingConvention = CallingConvention.Cdecl)]
|
public static extern IntPtr user_add_by_mat(IntPtr mat, string user_id, string group_id,
|
string user_info = "");
|
|
// 人脸更新(传图片帧)
|
[DllImport("BaiduFaceApi.dll", EntryPoint = "user_update", CharSet = CharSet.Ansi
|
, CallingConvention = CallingConvention.Cdecl)]
|
public static extern IntPtr user_update(IntPtr mat, string user_id, string group_id,
|
string user_info = "");
|
|
// 用户删除
|
[DllImport("BaiduFaceApi.dll", EntryPoint = "user_delete", CharSet = CharSet.Ansi
|
, CallingConvention = CallingConvention.Cdecl)]
|
public static extern IntPtr user_delete(string user_id, string group_id);
|
// 组添加
|
[DllImport("BaiduFaceApi.dll", EntryPoint = "group_add", CharSet = CharSet.Ansi
|
, CallingConvention = CallingConvention.Cdecl)]
|
public static extern IntPtr group_add(string group_id);
|
// 组删除
|
[DllImport("BaiduFaceApi.dll", EntryPoint = "group_delete", CharSet = CharSet.Ansi
|
, CallingConvention = CallingConvention.Cdecl)]
|
public static extern IntPtr group_delete(string group_id);
|
// 查询用户信息
|
[DllImport("BaiduFaceApi.dll", EntryPoint = "get_user_info", CharSet = CharSet.Ansi
|
, CallingConvention = CallingConvention.Cdecl)]
|
public static extern IntPtr get_user_info(string user_id, string group_id);
|
|
// 查询用户图片
|
[DllImport("BaiduFaceApi.dll", EntryPoint = "get_user_image", CharSet = CharSet.Ansi
|
, CallingConvention = CallingConvention.Cdecl)]
|
public static extern int get_user_image(IntPtr out_mat, string user_id, string group_id);
|
|
// 用户组列表查询
|
[DllImport("BaiduFaceApi.dll", EntryPoint = "get_user_list", CharSet = CharSet.Ansi
|
, CallingConvention = CallingConvention.Cdecl)]
|
public static extern IntPtr get_user_list(string group_id, int start = 0, int length = 100);
|
// 组列表查询
|
[DllImport("BaiduFaceApi.dll", EntryPoint = "get_group_list", CharSet = CharSet.Ansi
|
, CallingConvention = CallingConvention.Cdecl)]
|
public static extern IntPtr get_group_list(int start = 0, int length = 100);
|
|
// 数据库人脸数量查询
|
[DllImport("BaiduFaceApi.dll", EntryPoint = "db_face_count", CharSet = CharSet.Ansi
|
, CallingConvention = CallingConvention.Cdecl)]
|
public static extern int db_face_count(string group_id);
|
|
|
// 测试人脸注册
|
public void test_user_add()
|
{
|
// 人脸注册
|
string user_id = "test_user";
|
string group_id = "test_group";
|
string file_name = "../images/1.jpg";
|
|
string user_info = "user_info";
|
// 提取人脸特征值数组（多人会提取多个人的特征值）
|
/*
|
FaceFeature feature = new FaceFeature();
|
BDFaceFeature[] feaList1 = feature.test_get_face_feature_by_path(file_name
|
if (feaList1 == null)
|
{
|
Console.WriteLine("get feature fail");
|
return;
|
}
|
// 假设测试的图片是1个人，
|
BDFaceFeature f1 = feaList1[0];
|
// 人脸注册 (传特征值人脸注册，该方法注册不保存人脸图片入库)
|
IntPtr ptr = user_add(ref f1, user_id, group_id, user_info);
|
string buf = Marshal.PtrToStringAnsi(ptr);
|
Console.WriteLine("user_add res is:" + buf);
|
*/
|
// 通过图片帧人脸注册（只有该方法进行的人脸注册，人脸库才会保存人脸图片)
|
Mat mat = Cv2.ImRead(file_name);
|
IntPtr mptr = user_add_by_mat(mat.CvPtr, user_id, group_id, user_info);
|
string mbuf = Marshal.PtrToStringAnsi(mptr);
|
Console.WriteLine("user_add_by_mat res is:" + mbuf);
|
}
|
|
// 测试人脸更新
|
public void test_user_update()
|
{
|
string user_id = "test_user";
|
string group_id = "test_group";
|
string file_name = "../images/1.jpg";
|
Mat mat = Cv2.ImRead(file_name);
|
string user_info = "user_info";
|
// 人脸更新
|
IntPtr ptr = user_update(mat.CvPtr, user_id, group_id, user_info);
|
string buf = Marshal.PtrToStringAnsi(ptr);
|
Console.WriteLine("user_update res is:" + buf);
|
}
|
|
// 测试用户删除 （用户删除后，人脸数据也被删除）
|
public void test_user_delete()
|
{
|
string user_id = "test_user";
|
string group_id = "test_group";
|
IntPtr ptr = user_delete(user_id, group_id);
|
string buf = Marshal.PtrToStringAnsi(ptr);
|
Console.WriteLine("user_delete res is:" + buf);
|
}
|
|
// 组添加
|
public void test_group_add()
|
{
|
string group_id = "test_group2";
|
IntPtr ptr = group_add(group_id);
|
string buf = Marshal.PtrToStringAnsi(ptr);
|
Console.WriteLine("group_add res is:" + buf);
|
}
|
|
// 组删除
|
public void test_group_delete()
|
{
|
string group_id = "test_group2";
|
IntPtr ptr = group_delete(group_id);
|
string buf = Marshal.PtrToStringAnsi(ptr);
|
Console.WriteLine("group_delete res is:" + buf);
|
}
|
|
// 查询用户信息
|
public void test_get_user_info()
|
{
|
string user_id = "test_user";
|
string group_id = "test_group";
|
IntPtr ptr = get_user_info(user_id , group_id);
|
string buf = Marshal.PtrToStringAnsi(ptr);
|
Console.WriteLine("get_user_info res is:" + buf);
|
}
|
|
// 查询用户图片
|
public void test_get_user_image()
|
{
|
string user_id = "test_user";
|
string group_id = "test_group";
|
Mat out_mat = new Mat();
|
int res = get_user_image(out_mat.CvPtr, user_id, group_id);
|
if (res == 0)
|
{
|
Console.WriteLine("get_user_image success");
|
// 图片保存到本地
|
out_mat.ImWrite("user.jpg");
|
}
|
else
|
{
|
Console.WriteLine("get_user_image error{0}:", res);
|
}
|
|
}
|
|
// 用户组列表查询
|
public void test_get_user_list()
|
{
|
string group_id = "test_group";
|
IntPtr ptr = get_user_list(group_id);
|
string buf = Marshal.PtrToStringAnsi(ptr);
|
Console.WriteLine("get_user_list res is:" + buf);
|
}
|
|
// 组列表查询
|
public void test_get_group_list()
|
{
|
IntPtr ptr = get_group_list();
|
string buf = Marshal.PtrToStringAnsi(ptr);
|
Console.WriteLine("get_group_list res is:" + buf);
|
}
|
|
// 人脸库数量查询
|
public void test_db_face_count()
|
{
|
string group_id = "test_group";
|
// 参数传组id表示查该组都人脸数量
|
int count = db_face_count(group_id);
|
Console.WriteLine("count is:" + count);
|
string group_id2 = null;
|
// 参数传null表示查整个库
|
int count2 = db_face_count(group_id2);
|
Console.WriteLine("all count is:" + count2);
|
}
|
}
|
}
|
