7 changed files with 293 additions and 16 deletions
Binary file not shown.
@ -0,0 +1,107 @@ |
|||||
|
package yq.common.arcsoft; |
||||
|
|
||||
|
|
||||
|
import com.arcsoft.face.*; |
||||
|
import com.arcsoft.face.enums.DetectMode; |
||||
|
import com.arcsoft.face.enums.DetectOrient; |
||||
|
import com.arcsoft.face.enums.ErrorInfo; |
||||
|
import com.arcsoft.face.toolkit.ImageInfo; |
||||
|
import yq.common.utils.FaceEnginUtil; |
||||
|
|
||||
|
import java.io.File; |
||||
|
import java.util.ArrayList; |
||||
|
import java.util.List; |
||||
|
|
||||
|
import static com.arcsoft.face.toolkit.ImageFactory.getRGBData; |
||||
|
|
||||
|
|
||||
|
public class FaceEngineTest { |
||||
|
|
||||
|
|
||||
|
public static void main(String[] args) { |
||||
|
|
||||
|
// //从官网获取
|
||||
|
// String appId = "7MtFQZ9S9Ue2UC7koYxJbeL5iEHhqGtp3d5SJbJ4ou92";
|
||||
|
// String sdkKey = "4VM5rVL8eKmmELhRi8ZoopH33GzsgPoptA94zSXyHvzz";
|
||||
|
//
|
||||
|
//
|
||||
|
// FaceEngine faceEngine = new FaceEngine("E:\\arcsoft_lib\\LINUX64");
|
||||
|
// //激活引擎
|
||||
|
// int errorCode = faceEngine.activeOnline(appId, sdkKey);
|
||||
|
//
|
||||
|
// if (errorCode != ErrorInfo.MOK.getValue() && errorCode != ErrorInfo.MERR_ASF_ALREADY_ACTIVATED.getValue()) {
|
||||
|
// System.out.println("引擎激活失败");
|
||||
|
// }
|
||||
|
//
|
||||
|
//
|
||||
|
// ActiveFileInfo activeFileInfo=new ActiveFileInfo();
|
||||
|
// errorCode = faceEngine.getActiveFileInfo(activeFileInfo);
|
||||
|
// if (errorCode != ErrorInfo.MOK.getValue() && errorCode != ErrorInfo.MERR_ASF_ALREADY_ACTIVATED.getValue()) {
|
||||
|
// System.out.println("获取激活文件信息失败");
|
||||
|
// }
|
||||
|
//
|
||||
|
// //引擎配置
|
||||
|
// EngineConfiguration engineConfiguration = new EngineConfiguration();
|
||||
|
// engineConfiguration.setDetectMode(DetectMode.ASF_DETECT_MODE_IMAGE);
|
||||
|
// engineConfiguration.setDetectFaceOrientPriority(DetectOrient.ASF_OP_ALL_OUT);
|
||||
|
// engineConfiguration.setDetectFaceMaxNum(10);
|
||||
|
// engineConfiguration.setDetectFaceScaleVal(16);
|
||||
|
// //功能配置
|
||||
|
// FunctionConfiguration functionConfiguration = new FunctionConfiguration();
|
||||
|
// functionConfiguration.setSupportAge(true);
|
||||
|
// functionConfiguration.setSupportFace3dAngle(true);
|
||||
|
// functionConfiguration.setSupportFaceDetect(true);
|
||||
|
// functionConfiguration.setSupportFaceRecognition(true);
|
||||
|
// functionConfiguration.setSupportGender(true);
|
||||
|
// functionConfiguration.setSupportLiveness(true);
|
||||
|
// functionConfiguration.setSupportIRLiveness(true);
|
||||
|
// engineConfiguration.setFunctionConfiguration(functionConfiguration);
|
||||
|
//
|
||||
|
//
|
||||
|
// //初始化引擎
|
||||
|
// errorCode = faceEngine.init(engineConfiguration);
|
||||
|
//
|
||||
|
// if (errorCode != ErrorInfo.MOK.getValue()) {
|
||||
|
// System.out.println("初始化引擎失败");
|
||||
|
// }
|
||||
|
//
|
||||
|
//
|
||||
|
// //人脸检测
|
||||
|
// ImageInfo imageInfo = getRGBData(new File("E:\\dog\\dog1.jpg"));
|
||||
|
// List<FaceInfo> faceInfoList = new ArrayList<FaceInfo>();
|
||||
|
// errorCode = faceEngine.detectFaces(imageInfo.getImageData(), imageInfo.getWidth(), imageInfo.getHeight(), imageInfo.getImageFormat(), faceInfoList);
|
||||
|
// System.out.println(faceInfoList);
|
||||
|
//
|
||||
|
// //特征提取
|
||||
|
// FaceFeature faceFeature = new FaceFeature();
|
||||
|
// errorCode = faceEngine.extractFaceFeature(imageInfo.getImageData(), imageInfo.getWidth(), imageInfo.getHeight(), imageInfo.getImageFormat(), faceInfoList.get(0), faceFeature);
|
||||
|
// System.out.println("特征值大小：" + faceFeature.getFeatureData().length);
|
||||
|
//
|
||||
|
// //人脸检测2
|
||||
|
// ImageInfo imageInfo2 = getRGBData(new File("E:\\dog\\dog2.jpg"));
|
||||
|
// List<FaceInfo> faceInfoList2 = new ArrayList<FaceInfo>();
|
||||
|
// errorCode = faceEngine.detectFaces(imageInfo2.getImageData(), imageInfo2.getWidth(), imageInfo2.getHeight(),imageInfo2.getImageFormat(), faceInfoList2);
|
||||
|
// System.out.println(faceInfoList2);
|
||||
|
//
|
||||
|
// //特征提取2
|
||||
|
// FaceFeature faceFeature2 = new FaceFeature();
|
||||
|
// errorCode = faceEngine.extractFaceFeature(imageInfo2.getImageData(), imageInfo2.getWidth(), imageInfo2.getHeight(), imageInfo2.getImageFormat(), faceInfoList2.get(0), faceFeature2);
|
||||
|
// System.out.println("特征值大小：" + faceFeature2.getFeatureData().length);
|
||||
|
//
|
||||
|
// //特征比对
|
||||
|
// FaceFeature targetFaceFeature = new FaceFeature();
|
||||
|
// targetFaceFeature.setFeatureData(faceFeature.getFeatureData());
|
||||
|
// FaceFeature sourceFaceFeature = new FaceFeature();
|
||||
|
// sourceFaceFeature.setFeatureData(faceFeature2.getFeatureData());
|
||||
|
// FaceSimilar faceSimilar = new FaceSimilar();
|
||||
|
//
|
||||
|
// errorCode = faceEngine.compareFaceFeature(targetFaceFeature, sourceFaceFeature, faceSimilar);
|
||||
|
//
|
||||
|
// System.out.println("相似度：" + faceSimilar.getScore());
|
||||
|
//
|
||||
|
//
|
||||
|
// //引擎卸载
|
||||
|
// errorCode = faceEngine.unInit();
|
||||
|
|
||||
|
} |
||||
|
} |
||||
@ -0,0 +1,157 @@ |
|||||
|
package yq.common.utils; |
||||
|
|
||||
|
|
||||
|
import cn.hutool.core.collection.CollectionUtil; |
||||
|
import com.arcsoft.face.*; |
||||
|
import com.arcsoft.face.enums.DetectMode; |
||||
|
import com.arcsoft.face.enums.DetectOrient; |
||||
|
import com.arcsoft.face.enums.ErrorInfo; |
||||
|
import com.arcsoft.face.enums.ImageFormat; |
||||
|
import com.arcsoft.face.toolkit.ImageFactory; |
||||
|
import com.arcsoft.face.toolkit.ImageInfo; |
||||
|
import lombok.extern.slf4j.Slf4j; |
||||
|
import org.springframework.beans.factory.annotation.Value; |
||||
|
import org.springframework.web.multipart.MultipartFile; |
||||
|
import yq.common.exception.ServiceException; |
||||
|
import yq.common.utils.file.ImageUtils; |
||||
|
|
||||
|
import javax.imageio.ImageIO; |
||||
|
import java.awt.image.BufferedImage; |
||||
|
import java.io.IOException; |
||||
|
import java.io.InputStream; |
||||
|
import java.util.ArrayList; |
||||
|
import java.util.List; |
||||
|
|
||||
|
|
||||
|
/** |
||||
|
* minio文件上传工具类 |
||||
|
* @author: jeecg-boot |
||||
|
*/ |
||||
|
@Slf4j |
||||
|
public class FaceEnginUtil { |
||||
|
@Value(value = "${arcsoft.appId}") |
||||
|
private static String appId; |
||||
|
|
||||
|
@Value(value = "${arcsoft.sdkKey}") |
||||
|
private static String sdkKey; |
||||
|
|
||||
|
|
||||
|
@Value(value = "${arcsoft.faceEngine}") |
||||
|
private static String faceEngineUrl; |
||||
|
|
||||
|
|
||||
|
public static FaceEngine init() { |
||||
|
FaceEngine faceEngine = new FaceEngine(faceEngineUrl); |
||||
|
//激活引擎
|
||||
|
int errorCode = faceEngine.activeOnline(appId, sdkKey); |
||||
|
|
||||
|
if (errorCode != ErrorInfo.MOK.getValue() && errorCode != ErrorInfo.MERR_ASF_ALREADY_ACTIVATED.getValue()) { |
||||
|
System.out.println("引擎激活失败"); |
||||
|
} |
||||
|
|
||||
|
|
||||
|
ActiveFileInfo activeFileInfo=new ActiveFileInfo(); |
||||
|
errorCode = faceEngine.getActiveFileInfo(activeFileInfo); |
||||
|
if (errorCode != ErrorInfo.MOK.getValue() && errorCode != ErrorInfo.MERR_ASF_ALREADY_ACTIVATED.getValue()) { |
||||
|
System.out.println("获取激活文件信息失败"); |
||||
|
} |
||||
|
|
||||
|
//引擎配置
|
||||
|
EngineConfiguration engineConfiguration = new EngineConfiguration(); |
||||
|
engineConfiguration.setDetectMode(DetectMode.ASF_DETECT_MODE_IMAGE); |
||||
|
engineConfiguration.setDetectFaceOrientPriority(DetectOrient.ASF_OP_ALL_OUT); |
||||
|
engineConfiguration.setDetectFaceMaxNum(10); |
||||
|
engineConfiguration.setDetectFaceScaleVal(16); |
||||
|
//功能配置
|
||||
|
FunctionConfiguration functionConfiguration = new FunctionConfiguration(); |
||||
|
functionConfiguration.setSupportAge(true); |
||||
|
functionConfiguration.setSupportFace3dAngle(true); |
||||
|
functionConfiguration.setSupportFaceDetect(true); |
||||
|
functionConfiguration.setSupportFaceRecognition(true); |
||||
|
functionConfiguration.setSupportGender(true); |
||||
|
functionConfiguration.setSupportLiveness(true); |
||||
|
functionConfiguration.setSupportIRLiveness(true); |
||||
|
engineConfiguration.setFunctionConfiguration(functionConfiguration); |
||||
|
|
||||
|
|
||||
|
//初始化引擎
|
||||
|
errorCode = faceEngine.init(engineConfiguration); |
||||
|
|
||||
|
if (errorCode != ErrorInfo.MOK.getValue()) { |
||||
|
System.out.println("初始化引擎失败"); |
||||
|
} |
||||
|
return faceEngine; |
||||
|
} |
||||
|
|
||||
|
/** |
||||
|
* 录入人脸接口存入byte数组特征值 |
||||
|
* @param file |
||||
|
* @return |
||||
|
*/ |
||||
|
public static byte[] uploadInput(MultipartFile file) { |
||||
|
FaceEngine faceEngine = FaceEnginUtil.init(); |
||||
|
InputStream inputstream = null; |
||||
|
ImageInfo imageInfo = null; |
||||
|
try { |
||||
|
inputstream = file.getInputStream(); |
||||
|
imageInfo = ImageFactory.getRGBData(inputstream); |
||||
|
if (inputstream != null) { |
||||
|
inputstream.close(); |
||||
|
} |
||||
|
} catch (IOException e) { |
||||
|
throw new ServiceException("文件非图像类型"); |
||||
|
} |
||||
|
|
||||
|
List<FaceInfo> faceInfoList = new ArrayList<FaceInfo>(); |
||||
|
faceEngine.detectFaces(imageInfo.getImageData(), imageInfo.getWidth(), imageInfo.getHeight(), imageInfo.getImageFormat(), faceInfoList); |
||||
|
//特征提取
|
||||
|
FaceFeature faceFeature = new FaceFeature(); |
||||
|
faceEngine.extractFaceFeature(imageInfo.getImageData(), imageInfo.getWidth(), imageInfo.getHeight(), imageInfo.getImageFormat(), faceInfoList.get(0), faceFeature); |
||||
|
byte[] featureData = faceFeature.getFeatureData(); |
||||
|
FaceEnginUtil.close(faceEngine); |
||||
|
return featureData; |
||||
|
} |
||||
|
|
||||
|
/** |
||||
|
* 人脸匹配 |
||||
|
* @param file 人脸数据 |
||||
|
* @param faceFeature1 数据库人脸存根 |
||||
|
* @return float 数据判断大于0.7以上 |
||||
|
*/ |
||||
|
public static float uploadContrast(MultipartFile file,Byte[] faceFeature1) { |
||||
|
FaceEngine faceEngine = FaceEnginUtil.init(); |
||||
|
InputStream inputstream = null; |
||||
|
ImageInfo imageInfo = null; |
||||
|
try { |
||||
|
inputstream = file.getInputStream(); |
||||
|
imageInfo = ImageFactory.getRGBData(inputstream); |
||||
|
if (inputstream != null) { |
||||
|
inputstream.close(); |
||||
|
} |
||||
|
} catch (IOException e) { |
||||
|
throw new ServiceException("文件非图像类型"); |
||||
|
} |
||||
|
|
||||
|
List<FaceInfo> faceInfoList = new ArrayList<FaceInfo>(); |
||||
|
faceEngine.detectFaces(imageInfo.getImageData(), imageInfo.getWidth(), imageInfo.getHeight(), imageInfo.getImageFormat(), faceInfoList); |
||||
|
//特征提取
|
||||
|
FaceFeature faceFeature = new FaceFeature(); |
||||
|
faceEngine.extractFaceFeature(imageInfo.getImageData(), imageInfo.getWidth(), imageInfo.getHeight(), imageInfo.getImageFormat(), faceInfoList.get(0), faceFeature); |
||||
|
byte[] featureData = faceFeature.getFeatureData(); |
||||
|
//特征比对
|
||||
|
FaceFeature targetFaceFeature = new FaceFeature(); |
||||
|
targetFaceFeature.setFeatureData(featureData); |
||||
|
FaceFeature sourceFaceFeature = new FaceFeature(); |
||||
|
sourceFaceFeature.setFeatureData(faceFeature.getFeatureData()); |
||||
|
FaceSimilar faceSimilar = new FaceSimilar(); |
||||
|
faceEngine.compareFaceFeature(targetFaceFeature, sourceFaceFeature, faceSimilar); |
||||
|
FaceEnginUtil.close(faceEngine); |
||||
|
return faceSimilar.getScore(); |
||||
|
} |
||||
|
|
||||
|
public static void close(FaceEngine faceEngine) { |
||||
|
faceEngine.unInit(); |
||||
|
} |
||||
|
|
||||
|
|
||||
|
} |
||||
Loading…
Reference in new issue