/// <summary> /// 人脸比对 /// </summary> /// <param name="bitmap">输入图片</param> public static void FaceMatch(Bitmap bitmap) { var bmpData = bitmap.LockBits(new Rectangle(0, 0, bitmap.Width, bitmap.Height), ImageLockMode.ReadOnly, PixelFormat.Format24bppRgb); var imageData = new ImageData { PixelArrayFormat = 513,//Rgb24, Width = bitmap.Width, Height = bitmap.Height, Pitch = new int[4] { bmpData.Stride, 0, 0, 0 }, ppu8Plane = new IntPtr[4] { bmpData.Scan0, IntPtr.Zero, IntPtr.Zero, IntPtr.Zero } }; var ret = (ErrorCode)ArcWrapper.Detection(_DEnginer, ref imageData, out pDetectResult); if (ret != ErrorCode.Ok) { bitmap.UnlockBits(bmpData); return; } var detectResult = Marshal.PtrToStructure <DetectResult>(pDetectResult); CacheFaceResults.FaceNumber = detectResult.FaceCout; for (int i = detectResult.FaceCout; i < _MaxFaceNumber; i++) { CacheFaceResults[i].ID = ""; } if (detectResult.FaceCout == 0) { bitmap.UnlockBits(bmpData); return; } if (detectResult.FaceCout == 1) { CacheFaceResults[0].FFI.FaceRect = Marshal.PtrToStructure <FaceRect>(detectResult.PFaceRect); if (ArcWrapper.ExtractFeature(_REngine[0], ref imageData, ref CacheFaceResults[0].FFI, out fm) == (int)ErrorCode.Ok) { ArcWrapper.CopyMemory(CacheFaceResults.Items[0].FaceModel.PFeature, fm.PFeature, FeatureSize); } } else { Task[] ts = new Task[TaskNum < detectResult.FaceCout ? TaskNum : detectResult.FaceCout]; int faceOffset = -1; for (int i = 0; i < ts.Length; i++) { var rEngine = _REngine[i]; ts[i] = Task.Factory.StartNew(() => { int faceIndex = 0; while ((faceIndex = Interlocked.Increment(ref faceOffset)) < detectResult.FaceCout) { CacheFaceResults[faceIndex].FFI.FaceRect = Marshal.PtrToStructure <FaceRect>(IntPtr.Add(detectResult.PFaceRect, faceIndex * Marshal.SizeOf <FaceRect>())); if (ArcWrapper.ExtractFeature(rEngine, ref imageData, ref CacheFaceResults[faceIndex].FFI, out fm) == (int)ErrorCode.Ok) { ArcWrapper.CopyMemory(CacheFaceResults.Items[faceIndex].FaceModel.PFeature, fm.PFeature, FeatureSize); } } }); } Task.WaitAll(ts); } bitmap.UnlockBits(bmpData); var tsr = new Task[TaskNum]; List <int> noMatchFaces = Enumerable.Range(0, detectResult.FaceCout).ToList(); for (int i = 0; i < TaskNum; i++) { var taskIndex = i; tsr[i] = Task.Factory.StartNew(() => { var rEngine = _REngine[taskIndex]; foreach (var item in _FaceLib[taskIndex].Items.OrderByDescending(ii => ii.OrderId)) { _RWL.EnterReadLock(); if (noMatchFaces.Count == 0) { _RWL.ExitReadLock(); break; } var faceIndexs = noMatchFaces.ToList(); _RWL.ExitReadLock(); foreach (var faceIndex in faceIndexs) { ArcWrapper.Match(rEngine, ref CacheFaceResults.Items[faceIndex].FaceModel, ref item.FaceModel, out score); if (score > 0.55) { _RWL.EnterWriteLock(); noMatchFaces.Remove(faceIndex); _RWL.ExitWriteLock(); CacheFaceResults[faceIndex].ID = item.ID; CacheFaceResults[faceIndex].Score = score; item.OrderId = DateTime.Now.Ticks; } } } }); } Task.WaitAll(tsr); foreach (var faceIndex in noMatchFaces) { CacheFaceResults[faceIndex].ID = ""; CacheFaceResults[faceIndex].Score = 0; } }
/// <summary> /// 人脸比对 /// </summary> /// <param name="bitmap">输入图片</param> /// <param name="featureDataIndex">需要转换人脸特征的序号</param> public static void FaceMatch(Bitmap bitmap, int featureDataIndex = -1) { var bmpData = bitmap.LockBits(new Rectangle(0, 0, bitmap.Width, bitmap.Height), ImageLockMode.ReadOnly, PixelFormat.Format24bppRgb); var imageData = new ImageData { u32PixelArrayFormat = 513,//Rgb24, i32Width = bitmap.Width, i32Height = bitmap.Height, pi32Pitch = new int[4], ppu8Plane = new IntPtr[4] }; imageData.pi32Pitch[0] = bmpData.Stride; imageData.ppu8Plane[0] = bmpData.Scan0; try { var ret = (ErrorCode)ArcWrapper.Detection(_DEnginer, ref imageData, out var pDetectResult); if (ret != ErrorCode.Ok) { return; } var detectResult = Marshal.PtrToStructure <DetectResult>(pDetectResult); FaceResults.FaceNumber = detectResult.nFace; if (detectResult.nFace == 0) { return; } for (int i = 0; i < detectResult.nFace; i++) { IntPtr p = new IntPtr(detectResult.rcFace.ToInt32() + i * Marshal.SizeOf <FaceRect>()); var faceRect = Marshal.PtrToStructure <FaceRect>(p); FaceResults[i].Rectangle = new Rectangle(faceRect.left, faceRect.top, faceRect.right - faceRect.left, faceRect.bottom - faceRect.top); p = new IntPtr(detectResult.lfaceOrient.ToInt32() + i * Marshal.SizeOf <int>()); var faceOrient = Marshal.PtrToStructure <int>(p); var faceFeatureInput = new FaceFeatureInput { rcFace = faceRect, lOrient = faceOrient }; if ((ErrorCode)ArcWrapper.ExtractFeature(_REngine, ref imageData, ref faceFeatureInput, out var faceModel) != ErrorCode.Ok) { continue; } if (featureDataIndex == i) { if (FaceResults[i].FeatureData == null) { FaceResults[i].FeatureData = new byte[faceModel.lFeatureSize]; } Marshal.Copy(faceModel.pbFeature, FaceResults[i].FeatureData, 0, faceModel.lFeatureSize); } bool matched = false; foreach (var item in _FaceLib.Items.OrderByDescending(ii => ii.OrderId)) { var fm = item.FaceModel; ArcWrapper.Match(_REngine, ref fm, ref faceModel, out float score); if (score > 0.5) { matched = true; item.OrderId = DateTime.Now.Ticks; FaceResults[i].ID = item.ID; FaceResults[i].Score = score; break; } } if (!matched) { FaceResults[i].ID = "不认识"; FaceResults[i].Score = 0; } } } finally { bitmap.UnlockBits(bmpData); } }