git init

liuxiaolong

2019-05-06 f99bc8c6a1d10610373738edd7d0aa0181c81d99

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
package com.cloud.user.utils;
 
import java.io.FileInputStream;
import java.io.FileNotFoundException;
import java.io.IOException;
 
import org.apache.commons.io.IOUtils;
import org.opencv.core.Mat;
import org.opencv.core.MatOfByte;
import org.opencv.imgcodecs.Imgcodecs;
 
import com.cloud.common.exception.ApplicationException;
import com.zeroc.Ice.ConnectionRefusedException;
 
import ExtFaceWithImgOrUrl.ExtractFacePrx;
import ExtFaceWithImgOrUrl.InParameters;
import ExtFaceWithImgOrUrl.OutParameters;
import nu.pattern.OpenCV;
 
/**
 * 人脸工具类
 */
public class FaceUtil {
 
    private final String ST_FACE_SERVER = "ExtractFace:default -h 139.224.105.107 -p 10019";
    com.zeroc.Ice.Communicator communicator = com.zeroc.Ice.Util.initialize(new String[] { "/home" });
    private ExtractFacePrx stFaceServer = null;
 
    private ExtractFacePrx getServer() {
        if (stFaceServer != null) {
            return stFaceServer;
        } else {
            com.zeroc.Ice.ObjectPrx base = communicator.stringToProxy(ST_FACE_SERVER);
            stFaceServer = ExtractFacePrx.checkedCast(base);
            return stFaceServer;
        }
    }
 
    /**
     * 传入文件的字节数组， 返回人脸的feature
     * @param byteArray
     * @param imageUrl
     * @return
     */
    public static String extractFeature(byte[] byteArray, String imageUrl) {
        FaceUtil client = new FaceUtil();
        try {
            OpenCV.loadShared();
            ExtractFacePrx server = client.getServer();
            if (server != null) {
                if (byteArray != null) {
                    Mat image = Imgcodecs.imdecode(new MatOfByte(byteArray), Imgcodecs.CV_LOAD_IMAGE_UNCHANGED);
                    MatOfByte buffer = new MatOfByte();
                    Imgcodecs.imencode(".jpg", image, buffer);
                    byte[] array = buffer.toArray();
                    InParameters inParm = new InParameters(array, imageUrl);
                    OutParameters outParm = server.extractFaceWithImgOrUrl(inParm);
                    return outParm.faceFeaBase64;
                } else if (imageUrl != null && imageUrl.length() > 0) {
//                    Mat image = Imgcodecs.imdecode(new MatOfByte(byteArray), Imgcodecs.CV_LOAD_IMAGE_UNCHANGED);
//                    MatOfByte buffer = new MatOfByte();
//                    Mat image =Imgcodecs.imread("/home/bsk7/图片/image/6.jpg");
//                    Imgcodecs.imencode(".jpg", image, buffer);
//                    byte[] array = buffer.toArray();
//                    InParameters inParm = new InParameters(array, imageUrl);
//                    OutParameters outParm = server.extractFaceWithImgOrUrl(inParm);
//                    return outParm.faceFeaBase64;
                }
            }
        } catch (ConnectionRefusedException e) {
            throw new ApplicationException("连接人脸识别服务器失败.");
        } catch (Exception e) {
            e.printStackTrace();
        } finally {
            client.communicator.destroy();
        }
        return null;
    }
 
    public static void main(String[] args) {
        try {
            String ret = extractFeature(IOUtils.toByteArray(new FileInputStream("d:\\7_1.jpg")), null);
            System.out.println(ret);
        } catch (Exception e) {
            e.printStackTrace();
        }
    }
    
}