Apache Servicecomb結合人臉識別API實現人臉識別

點擊上面“藍字”關注我們！

本篇文章主要講述如何在本地代碼中調用華爲雲產品人臉識別服務API，實現人臉識別功能。

完整Demo參考：https://github.com/servicestage-demo/cse-java-demo-list/tree/master/demo-cse-face-recognition

一、創建工程

1、首先在本地搭建ServiceComb工程

爲了能夠使開發者可以快速構建ServiceComb應用程序，官方爲我們提供了一套腳手架，這樣能夠方便學習者及應用開發者快速入門，同時極大的提高了效率。

快速開發引導頁：http://start.servicecomb.io/

填寫好工程group、artifact等信息（ServiceComb參數可以使用默認），就可以生成一個ServiceComb工程了，將工程下載到本地，導入到Eclipse或者IDEA等開發工具中。

2、配置microservice.yaml

servicecomb.service.registry.address：CSE（微服務引擎）服務註冊發現地址

servicecomb.rest.address：本地應用訪問地址

AK、SK獲取請參考鏈接：https://support.huaweicloud.com/devg-apisign/api-sign-provide.html#p3

APPLICATION_ID: demo-face-recognition
service_description:
# name of the declaring microservice
  name: demo-face-recognition
  version: 0.0.1
  environment: development
servicecomb:
  service:
    registry:
      address: https://cse.xxx.myhuaweicloud.com:port
  rest:
    address: 0.0.0.0:8081
  credentials:
    accessKey: Your AK
    secretKey: Your SK
    project: cn-north-1
    akskCustomerCipher: default
  handler:
    chain:
      Provider:
        default: tracing-provider

也可以使用本地註冊中心，到官網http://servicecomb.apache.org/cn/release/下載ServiceComb Service-Center（選最新版本）

本地microservice.yaml配置參考：

APPLICATION_ID: demo-face-recognition
service_description:
  name: demo-face-recognition
  version: 1.0.0
service_description:
  name: demo-provider
  version: 1.0.0
servicecomb:
  rest:
    address: 0.0.0.0:9000
  service:
    registry:
      address: http://127.0.0.1:30100

3、配置application.properties（與microservice.yaml同目錄）

#圖片保存路徑：自定義

#獲取token參數：參考Token認證鑑權指南https://support.huaweicloud.com/api-modelarts/modelarts_03_0004.html

#搜索人臉API：參考華爲人臉識別服務 FRS：https://support.huaweicloud.com/face/index.html

首先是申請開通服務（須註冊華爲雲賬號，並完成實名認證），照着下面這個地址一步步操作就行，比較詳細。

申請服務：https://support.huaweicloud.com/api-face/face_02_0006.html

然後要創建自己的人臉庫，參考：https://support.huaweicloud.com/api-face/face_02_0031.html

人臉庫擴展字段external_fields可以自定義，比如本文Demo設置了：name、gender、department等字段

#增加人臉API、#刪除人臉API同上。

#圖片保存路徑
face_img_path=/usr/gc/img/
#獲取token參數
face_token_url=https://xxx.myhuaweicloud.com/v3/auth/tokens
face_token_username=username
face_token_password=password
face_token_domain=domain
face_token_project=project
#搜索人臉API
face_api_search=https://face.xxx.myhuaweicloud.com/v1/xxx/face-sets/test/search
#增加人臉API
face_api_add=https://face.xxx.myhuaweicloud.com/v1/xxx/face-sets/test/faces
#刪除人臉API
face_api_del=https://face.xxx.myhuaweicloud.com/v1/xxx/face-sets/test/faces?external_image_id=imgID

4、定義接口

定義了搜索、增加、刪除3個接口，另外有個更新接口，可自行實現。

public interface FaceEndpoint {
  /**
   * search face
   */
  String face(MultipartFile file);




  /**
   * add face
   */
  String addface(MultipartFile file, String id);




  /**
   * delete face
   */
  String delface(String id);
}

5、編寫業務類

訪問路徑可以自定義，注意參數類型是file（直接上傳圖片進行識別）

@RestSchema(schemaId = "FaceRestEndpoint")
@RequestMapping(path = "/")
public class FaceRestEndpoint implements FaceEndpoint {
  private final FileService fileService;




  @Autowired
  public FaceRestEndpoint(FileService fileService) {
    this.fileService = fileService;
  }




  @PostMapping(path = "/face", produces = MediaType.TEXT_PLAIN_VALUE)
  public String face(@RequestPart(name = "file") MultipartFile file) {
    return fileService.face(file);
  }




  @PostMapping(path = "/addface", produces = MediaType.TEXT_PLAIN_VALUE)
  public String addface(@RequestPart(name = "file") MultipartFile file, @RequestParam(name = "id") String id) {
    return fileService.addface(file, id);
  }




  @DeleteMapping(path = "/delface", produces = MediaType.TEXT_PLAIN_VALUE)
  public String delface(String id) {
    return fileService.delface(id);
  }
}

6、實現獲取Token，調用人臉識別API

爲避免頻繁調用獲取Token接口，可以在本地緩存Token，Token有效期是24小時

package org.apache.servicecomb.samples.face;




import java.io.File;
import java.nio.charset.Charset;
import java.util.HashMap;
import java.util.List;
import java.util.Map;




import org.apache.log4j.Logger;
import org.apache.servicecomb.provider.springmvc.reference.RestTemplateBuilder;
import org.apache.servicecomb.tracing.Span;
import org.springframework.beans.factory.annotation.Value;
import org.springframework.core.io.FileSystemResource;
import org.springframework.http.HttpEntity;
import org.springframework.http.HttpHeaders;
import org.springframework.http.HttpMethod;
import org.springframework.http.MediaType;
import org.springframework.http.ResponseEntity;
import org.springframework.http.converter.HttpMessageConverter;
import org.springframework.http.converter.StringHttpMessageConverter;
import org.springframework.stereotype.Service;
import org.springframework.util.LinkedMultiValueMap;
import org.springframework.util.MultiValueMap;
import org.springframework.web.client.RestTemplate;
import org.springframework.web.multipart.MultipartFile;




import com.alibaba.fastjson.JSONArray;
import com.alibaba.fastjson.JSONObject;




@Service
public class FileServiceImpl implements FileService {
  @Value("${face_img_path}")
  private String faceImgPath;
  @Value("${face_token_url}")
  private String faceTokenUrl;
  @Value("${face_token_username}")
  private String faceTokenUsername;
  @Value("${face_token_password}")
  private String faceTokenPassword;
  @Value("${face_token_domain}")
  private String faceTokenDomain;
  @Value("${face_token_project}")
  private String faceTokenProject;
  @Value("${face_api_search}")
  private String faceApiSearch;
  @Value("${face_api_add}")
  private String faceApiAdd;
  @Value("${face_api_del}")
  private String faceApiDel;




  private static final String TOKEN_KEY = "X-Subject-Token";
  private static final String EXPIRES_KEY = "Expires-time";
  private static final long availablePeriod = 20 * 3600 * 1000; // 20小時




  private static Map<String, String> tokenMap = new HashMap<String, String>();
  private static Map<String, String> data = new HashMap<String, String>();




  private static Logger LOGGER = Logger.getLogger(FileServiceImpl.class);




  private String getToken() {
    LOGGER.info("start to get token");
    if (!tokenMap.isEmpty()) {
      long startTime = Long.valueOf(tokenMap.get(EXPIRES_KEY));
      long currTime = System.currentTimeMillis();
      if (currTime - startTime < availablePeriod) {
        return tokenMap.get(TOKEN_KEY);
      }
    }




    String raw = requestBody(faceTokenUsername, faceTokenPassword, faceTokenDomain, faceTokenProject);
    RestTemplate restTemplate = RestTemplateBuilder.create();
    HttpHeaders headers = new HttpHeaders();
    headers.set("Content-Type", "application/json");




    HttpEntity<String> entity = new HttpEntity<String>(raw, headers);
    ResponseEntity<String> r = restTemplate.postForEntity(faceTokenUrl, entity, String.class);
    String token = r.getHeaders().getFirst(TOKEN_KEY);
    tokenMap.put(TOKEN_KEY, token);
    tokenMap.put(EXPIRES_KEY, String.valueOf(System.currentTimeMillis()));
    LOGGER.info("end to get token");
    return token;
  }




  /**
   * 構造使用Token方式訪問服務的請求Token對象
   * 
   * @param username    用戶名
   * @param passwd      密碼
   * @param domainName  域名
   * @param projectName 項目名稱
   * @return 構造訪問的JSON對象
   */
  private static String requestBody(String username, String passwd, String domainName, String projectName) {
    JSONObject auth = new JSONObject();




    JSONObject identity = new JSONObject();




    JSONArray methods = new JSONArray();
    methods.add("password");
    identity.put("methods", methods);




    JSONObject password = new JSONObject();




    JSONObject user = new JSONObject();
    user.put("name", username);
    user.put("password", passwd);




    JSONObject domain = new JSONObject();
    domain.put("name", domainName);
    user.put("domain", domain);




    password.put("user", user);




    identity.put("password", password);




    JSONObject scope = new JSONObject();




    JSONObject scopeProject = new JSONObject();
    scopeProject.put("name", projectName);




    scope.put("project", scopeProject);




    auth.put("identity", identity);
    auth.put("scope", scope);




    JSONObject params = new JSONObject();
    params.put("auth", auth);
    return params.toJSONString();
  }




    //請求響應編碼轉換，防止亂碼
  public static RestTemplate getRestTemplate(String charset) {
    RestTemplate restTemplate = new RestTemplate();
    List<HttpMessageConverter<?>> list = restTemplate.getMessageConverters();
    for (HttpMessageConverter<?> httpMessageConverter : list) {
      if (httpMessageConverter instanceof StringHttpMessageConverter) {
        ((StringHttpMessageConverter) httpMessageConverter).setDefaultCharset(Charset.forName(charset));
        break;
      }
    }
    return restTemplate;
  }




  @Span
  @Override
  public String face(MultipartFile file) {
    // 此處構造數據，建議使用數據庫
    if (data.isEmpty()) {
      data.put("xiaoming", "{\"姓名\":\"小明\", \"性別\":\"男\", \"部門\":\"華爲雲服務\"}");
      data.put("unknow", "{\"姓名\":\"未知\", \"性別\":\"未知\", \"部門\":\"未知\"}");
    }




    LOGGER.info("start to face recognition");




    String filePath = null;
    if (file != null && !file.isEmpty()) {
      // 文件保存路徑
      filePath = faceImgPath + System.currentTimeMillis() + "-" + file.getOriginalFilename();
      // 轉存文件
      try {
        file.transferTo(new File(filePath));
      } catch (Exception e) {
        return e.getMessage();
      }
    }




    if (filePath == null) {
      return data.get("unknow");
    }




    RestTemplate restTemplate = getRestTemplate("UTF-8");




    // 設置請求頭
    HttpHeaders headers = new HttpHeaders();
    MediaType type = MediaType.parseMediaType("multipart/form-data;charset=UTF-8;");
    headers.setContentType(type);
    headers.set("X-Auth-Token", getToken());




    // 設置請求體，注意是LinkedMultiValueMap
    MultiValueMap<String, Object> form = new LinkedMultiValueMap<String, Object>();
    FileSystemResource fileSystemResource = new FileSystemResource(filePath);




    // 此處參數參考https://support.huaweicloud.com/api-face/face_02_0035.html
    form.add("image_file", fileSystemResource);
    form.add("return_fields", "[\"timestamp\",\"id\"]");
    form.add("filter", "timestamp:12");




    // 用HttpEntity封裝整個請求報文
    HttpEntity<MultiValueMap<String, Object>> files = new HttpEntity<MultiValueMap<String, Object>>(form, headers);
    ResponseEntity<String> s;
    try {
      s = restTemplate.postForEntity(faceApiSearch, files, String.class);
    } catch (Exception e) {
      return data.get("unknow");
    }




    LOGGER.info("end to face recognition");




    // 此處進行相似度判斷，實現邏輯可自行修改
    String name = s.getBody().substring(s.getBody().indexOf("external_image_id") + 20);
    name = name.substring(0, 3).replace("\"", "");
    String similar = s.getBody().substring(s.getBody().indexOf("similarity") + 12);
    similar = similar.substring(0, similar.indexOf("}"));
    // 相似度大於86%，則認爲人臉識別成功
    if (Double.parseDouble(similar) > 0.86) {
      return data.get(name);
    }
    return data.get("unknow");
  }




  @Override
  public String addface(MultipartFile file, String external_image_id) {
    LOGGER.info("start to addface file");




    String filePath = null;
    String imgName = System.currentTimeMillis() + "-" + file.getOriginalFilename();
    if (file != null && !file.isEmpty()) {
      // 文件保存路徑
      filePath = faceImgPath + imgName;
      // 轉存文件
      try {
        file.transferTo(new File(filePath));
      } catch (Exception e) {
        return e.getMessage();
      }
    }




    // 文件保存失敗，返回圖片路徑
    if (filePath == null) {
      return "{\"image\": \"" + (faceImgPath + imgName) + "\"}";
    }




    RestTemplate restTemplate = getRestTemplate("UTF-8");




    // 設置請求頭
    HttpHeaders headers = new HttpHeaders();
    MediaType type = MediaType.parseMediaType("multipart/form-data;charset=UTF-8;");
    headers.setContentType(type);
    headers.set("X-Auth-Token", getToken());




    // 設置請求體，注意是LinkedMultiValueMap
    MultiValueMap<String, Object> form = new LinkedMultiValueMap<String, Object>();
    FileSystemResource fileSystemResource = new FileSystemResource(filePath);




    // 此處參數參考https://support.huaweicloud.com/api-face/face_02_0035.html
    form.add("image_file", fileSystemResource);
    form.add("external_image_id", external_image_id);
    form.add("external_fields", "{\"timestamp\" : 12,\"id\" : \"home\"}");




    // 用HttpEntity封裝整個請求報文
    HttpEntity<MultiValueMap<String, Object>> files = new HttpEntity<MultiValueMap<String, Object>>(form, headers);
    ResponseEntity<String> s = restTemplate.postForEntity(faceApiAdd, files, String.class);
    LOGGER.info("end to addface file");
    return s.getBody();
  }




  @Override
  public String delface(String id) {
    RestTemplate restTemplate = getRestTemplate("UTF-8");




    // 設置請求頭
    HttpHeaders headers = new HttpHeaders();
    MediaType type = MediaType.parseMediaType("application/json;charset=UTF-8;");
    headers.setContentType(type);
    headers.set("X-Auth-Token", getToken());
    Map<String, String> paramMap = new HashMap<String, String>();
    paramMap.put("X-Auth-Token", getToken());




    // 用HttpEntity封裝整個請求報文
    String URL_HTTPS = faceApiDel + id;
    HttpEntity<String> requestEntity = new HttpEntity<String>(null, headers);
    ResponseEntity<String> s = restTemplate.exchange(URL_HTTPS, HttpMethod.DELETE, requestEntity, String.class);




    LOGGER.info("end to delface file");
    return s.getBody();
  }
}

7、調用示例

#搜索人臉接口

請求URL：http://xxx.xxx.213.158:8081/face
請求方式：POST
參數類型：body
參數示例：form-data (key:file, value:[圖片文件])
返回示例：{"姓名":"小明", "性別":"男", "部門":"華爲雲服務"}

#增加人臉接口

請求URL：http://xxx.xxx.213.158:8081/add
請求方式：POST
參數類型：body
參數示例：raw(json)
{
    "name":"谷歌",
    "gender":"女",
    "department":"雲服務"
}
返回示例：1

#刪除人臉接口

請求URL：http://xxx.xxx.213.158:8081/delface?id=xxx
請求方式：DELETE
參數類型：query
參數示例：見url (或者key:id, value:xxx)
返回示例：{"face_number":1, "face_set_name":"test", "face_set_id":"xxx"}

8、pom.xml參考

<?xml version="1.0" encoding="UTF-8"?>
<project xmlns="http://maven.apache.org/POM/4.0.0"
  xmlns:xsi="http://www.w3.org/2001/XMLSchema-instance"
  xsi:schemaLocation="http://maven.apache.org/POM/4.0.0 http://maven.apache.org/xsd/maven-4.0.0.xsd">
  <modelVersion>4.0.0</modelVersion>
  <groupId>org.apache.servicecomb.samples</groupId>
  <artifactId>garbage-classify</artifactId>
  <name>Java Chassis::Samples::Garbage-Classify</name>
  <version>1.2.0</version>
  <packaging>jar</packaging>




  <description>Quick Start Demo for Using ServiceComb Java Chassis</description>




  <properties>
    <project.build.sourceEncoding>UTF-8</project.build.sourceEncoding>
    <project.reporting.outputEncoding>UTF-8</project.reporting.outputEncoding>
    <java.version>1.8</java.version>
    <java-chassis.version>${project.version}</java-chassis.version>
    <spring-boot-1.version>1.5.14.RELEASE</spring-boot-1.version>
  </properties>




  <dependencyManagement>
    <dependencies>
      <dependency>
        <groupId>org.apache.servicecomb</groupId>
        <artifactId>java-chassis-dependencies</artifactId>
        <version>${java-chassis.version}</version>
        <type>pom</type>
        <scope>import</scope>
      </dependency>
      <dependency>
        <groupId>org.springframework.boot</groupId>
        <artifactId>spring-boot-dependencies</artifactId>
        <version>${spring-boot-1.version}</version>
        <type>pom</type>
        <scope>import</scope>
      </dependency>
    </dependencies>
  </dependencyManagement>




  <dependencies>
    <dependency>
      <groupId>org.hibernate.validator</groupId>
      <artifactId>hibernate-validator</artifactId>
    </dependency>
    <dependency>
      <groupId>javax.validation</groupId>
      <artifactId>validation-api</artifactId>
    </dependency>
    <dependency>
      <groupId>org.springframework.boot</groupId>
      <artifactId>spring-boot-starter</artifactId>
    </dependency>




    <dependency>
      <groupId>org.apache.servicecomb</groupId>
      <artifactId>spring-boot-starter-provider</artifactId>
    </dependency>




    <dependency>
      <groupId>org.apache.servicecomb</groupId>
      <artifactId>handler-flowcontrol-qps</artifactId>
    </dependency>
    <dependency>
      <groupId>org.apache.servicecomb</groupId>
      <artifactId>handler-bizkeeper</artifactId>
    </dependency>
    <dependency>
      <groupId>org.apache.servicecomb</groupId>
      <artifactId>handler-tracing-zipkin</artifactId>
    </dependency>
    <dependency>
      <groupId>com.huawei.paas.cse</groupId>
      <artifactId>cse-solution-service-engine</artifactId>
      <version>2.3.56</version>
    </dependency>
    <dependency>
      <groupId>org.apache.servicecomb</groupId>
      <artifactId>tracing-zipkin</artifactId>
    </dependency>
    <dependency>
      <groupId>org.mybatis</groupId>
      <artifactId>mybatis</artifactId>
      <version>3.4.1</version>
    </dependency>




    <dependency>
      <groupId>com.alibaba</groupId>
      <artifactId>druid</artifactId>
      <version>1.0.14</version>
    </dependency>




    <dependency>
      <groupId>com.alibaba</groupId>
      <artifactId>fastjson</artifactId>
      <version>1.2.31</version>
    </dependency>
  </dependencies>
  <build>
    <plugins>
      <plugin>
        <artifactId>maven-dependency-plugin</artifactId>
        <executions>
          <execution>
            <id>copy-dependencies</id>
            <phase>prepare-package</phase>
            <goals>
              <goal>copy-dependencies</goal>
            </goals>
            <configuration>
              <outputDirectory>
                ${project.build.directory}/dtm-lib
              </outputDirectory>
            </configuration>
          </execution>
        </executions>
      </plugin>
      <plugin>
        <artifactId>maven-jar-plugin</artifactId>
        <configuration>
          <outputDirectory>${project.build.directory}</outputDirectory>
          <archive>
            <manifest>
              <addClasspath>true</addClasspath>
              <classpathPrefix>./dtm-lib/</classpathPrefix>
              <mainClass>org.apache.servicecomb.samples.gc.GCApplication</mainClass>
              <useUniqueVersions>false</useUniqueVersions>
            </manifest>
            <manifestEntries>
              <Class-Path>.</Class-Path>
            </manifestEntries>
          </archive>
        </configuration>
      </plugin>
      <plugin>
        <groupId>org.apache.maven.plugins</groupId>
        <artifactId>maven-compiler-plugin</artifactId>
        <version>3.1</version>
        <configuration>
          <compilerArgument>-parameters</compilerArgument>
          <encoding>UTF-8</encoding>
          <source>1.8</source>
          <target>1.8</target>
        </configuration>
      </plugin>
      <plugin>
        <groupId>org.springframework.boot</groupId>
        <artifactId>spring-boot-maven-plugin</artifactId>
        <configuration>
          <fork>true</fork>
        </configuration>
      </plugin>
    </plugins>
  </build>
</project>

如您對開源開發，微服務專家

歡迎掃描下方二維碼添加

ServiceComb小助手

咱們一起做點有意思的事情〜

您點的每個贊，我都認真當成了喜歡