我的明星脸-人脸识别DEMO

public IdentifyImageResponse identifyImageInGroup(IdentifyImageRequest request, String appId, String groupName, String imageInfo, boolean isBosPath) {
    // json请求参数
    StringWriter writer = new StringWriter();
    try {
        JsonGenerator jsonGenerator = JsonUtils.jsonGeneratorOf(writer);
        jsonGenerator.writeStartObject();
        if(isBosPath) {
            jsonGenerator.writeStringField("bosPath", imageInfo);
        }
        else {
            jsonGenerator.writeStringField("base64", imageInfo);
        }
        jsonGenerator.writeEndObject();
        jsonGenerator.close();
    } catch (IOException e) {
        throw new BceClientException("Fail to generate json", e);
    }

    byte[] json = null;
    try {
        json = writer.toString().getBytes(DEFAULT_ENCODING);
    } catch (UnsupportedEncodingException e) {
        throw new BceClientException("Fail to get UTF-8 bytes", e);
    }

    InternalRequest internalRequest =
            this.createRequest(request, HttpMethodName.POST, APPSTR, appId, GROUPSTR, groupName);
    internalRequest.addParameter(IDENTIFYSTR, null);
    internalRequest.addHeader(Headers.CONTENT_LENGTH, String.valueOf(json.length));
    internalRequest.setContent(RestartableInputStream.wrap(json));

    IdentifyImageResponse response = this.invokeHttpClient(internalRequest, IdentifyImageResponse.class);
    return response;
}

public static Map<String, String> detectStarFace(BfrClient bfrClient, String appId, String imgInfo, boolean isBosPath) {
    Map<String, String> result = new HashMap<String, String>();
    result.clear();

    Map<String, Float> manMap = new ConcurrentHashMap<String, Float>();
    manMap.clear();
    Map<String, Float> womanMap = new ConcurrentHashMap<String, Float>();
    womanMap.clear();


    ListAllGroupsResponse listAllGroupsResponse = bfrClient.listAllGroups(BfrFaceUtil.APP_ID);
    List<GroupModel> groups = listAllGroupsResponse.getGroups();
    int consumeTime = 0;
    int manGroupNum = 0;
    int womanGroupNum = 0;
    for(int i = 0; i < groups.size(); i++) {
        String groupName = groups.get(i).getGroupName();
        if(groupName.contains(BfrFaceUtil.BFR_WOMANSTAR_GROUP_PREFIX)) {
            // 女组
            FixedThreadPool.executor.execute(new BfrFaceIdentifyThread(womanMap, bfrClient, appId, groupName, imgInfo, isBosPath));
            womanGroupNum++;
        }
        else {
            // 男组
            FixedThreadPool.executor.execute(new BfrFaceIdentifyThread(manMap, bfrClient, appId, groupName, imgInfo, isBosPath));
            manGroupNum++;
        }
    }
    while(manMap.size() < manGroupNum && womanMap.size() < womanGroupNum && consumeTime < 5000) {
        try {
            Thread.sleep(500);
            consumeTime += 500;
        }
        catch (Exception e) {
            logger.error("Detect star face Thread.sleep() occurs erorr: " + e);
        }
    }
    logger.info("Do detect star face consume time: " + consumeTime + "ms");

    if(manMap.isEmpty() || womanMap.isEmpty()) {
        result.put("msg", "FAILED");
    }
    else {
        result.put("msg", "OK");
        
        Map<String, Float> sortedManMap = MapUtil.sortByValue(manMap, false);
        Person manStar = FaceUtil.allStarMap.get(FaceUtil.person_id_prefix + sortedManMap.keySet().toArray()[0].toString());
        result.put("manName", manStar.getName());
        result.put("manImageUrl", manStar.getPicUrl());
        result.put("manConf", String.valueOf(sortedManMap.get(sortedManMap.keySet().toArray()[0].toString())));

        Map<String, Float> sortedWomanMap = MapUtil.sortByValue(womanMap, false);
        Person womanStar = FaceUtil.allStarMap.get(FaceUtil.person_id_prefix + sortedWomanMap.keySet().toArray()[0].toString());
        result.put("womanName", womanStar.getName());
        result.put("womanImageUrl", womanStar.getPicUrl());
        result.put("womanConf", String.valueOf(sortedWomanMap.get(sortedWomanMap.keySet().toArray()[0].toString())));
    }

    return result;
}