微软认知服务-Computer Vision API调用合集

package com.ictr.utils;

import java.io.File;

import java.io.InputStream;

import java.math.BigDecimal;

import java.net.URI;

import org.apache.http.HttpEntity;

import org.apache.http.HttpResponse;

import org.apache.http.client.HttpClient;

import org.apache.http.client.methods.HttpGet;

import org.apache.http.client.methods.HttpPost;

import org.apache.http.client.utils.URIBuilder;

import org.apache.http.entity.FileEntity;

import org.apache.http.entity.StringEntity;

import org.apache.http.impl.client.HttpClients;

import org.apache.http.util.EntityUtils;

import com.alibaba.fastjson.JSONArray;

import com.alibaba.fastjson.JSONObject;

/**

 * 这个类主要是微软认知服务中的计算机视觉

 * 

 * @author liuyb

 *

 */

public class CognitiveUtils {
public static final String KEY = "申请的微软的key";

// 图像分析的链接
public static final String ANALYZE_IMAGE = "https://api.projectoxford.ai/vision/v1.0/analyze";
// 描述图像的链接
public static final String DESCRIBE_IMAGE = "https://api.projectoxford.ai/vision/v1.0/describe";
// 获取缩略图的链接
public static final String GET_THUMBNAIL = "https://api.projectoxford.ai/vision/v1.0/generateThumbnail";

// 特定领域模型识别(现在只支持名人识别，这个常量支持两个接口)
public static final String Domain_SPECIFIC = "https://api.projectoxford.ai/vision/v1.0/models";

// 识别图片中的文字
public static final String OCR = "https://api.projectoxford.ai/vision/v1.0/ocr";

// 获取图片的标签
public static final String IMAGE_TAG = "https://api.projectoxford.ai/vision/v1.0/tag";

public static HttpClient httpclient = HttpClients.createDefault();

// 调用微软认知服务获取数据

/**
* 分析图片的接口
* 
* @param filePath
*/
public static JSONObject analyzeImage(String filePath) {
try {
URIBuilder builder = new URIBuilder(ANALYZE_IMAGE);

// 为链接配置的一些参数
/*
* 图像特征点： Categories:种类 ImageType:图片类型 Faces:人脸描述 Adult：监测成人内容
* Color:颜色 Tags:标签 Description:用一句完成的英文描述图片
*/
builder.setParameter("visualFeatures", "Description");
/*
* 对一些特定领域的细节的返回
* 
* 目前只支持“名人”识别:Celebrities
*/
// builder.setParameter("details", "{string}");
/*
* 以哪种语言返回，默认为英语 en 英文 zh 中文
*/
// builder.setParameter("language", "en");

URI uri = builder.build();
HttpPost request = new HttpPost(uri);
/**
* 请求类型 json application/json 图片以url方式发送 二进制流
* application/octet-stream 图片以二进制流发送 多文件发送 multipart/form-data
* 图片以二进制流进行发送
*/
// request.setHeader("Content-Type", "application/json");
// 以json方式访问，图片路径需要为一个能访问到的地址
request.setHeader("Content-Type", "application/octet-stream"); // 以位进制文件流的方式访问，图片需要转化成一个二进制的流

// 还有一种请求方式multipart/form-data

/*
* 请求API的秘钥（需要去微软官方申请）
*/
request.setHeader("Ocp-Apim-Subscription-Key", KEY);

File file = new File(filePath);

if (!file.exists()) {
System.out.println("图片不存在");
return null;
}

// Request body
// 以json方式访问时传递的是一个json字符串
// StringEntity reqEntity = new StringEntity("{\"Url\":
// \""+filePath+"\"}");

// 以文件流方式访问时传递的是一个文件（最终会以位二进制进行请求）
FileEntity reqEntity = new FileEntity(file);
request.setEntity(reqEntity);

HttpResponse response = httpclient.execute(request);
HttpEntity entity = response.getEntity();

if (entity != null) {
String result = EntityUtils.toString(entity);

System.out.println();

return (JSONObject) JSONObject.parse(result);
}
} catch (Exception e) {
System.out.println(e.getMessage());
}

return null;
}

/**
* 描述图片
* 
* @param filePath
*/
public static void describeImage(String filePath) {
try {
URIBuilder builder = new URIBuilder(DESCRIBE_IMAGE);

// 最大数量的候选人描述返回。缺省值是1。如果是1，则返回可信度最高
e3e5
的结果
builder.setParameter("maxCandidates", "1");

URI uri = builder.build();
HttpPost request = new HttpPost(uri);

// application/json application/octet-stream multipart/form-data
request.setHeader("Content-Type", "application/json");
request.setHeader("Ocp-Apim-Subscription-Key", KEY);

// Request body
StringEntity reqEntity = new StringEntity("{\"Url\": \"" + filePath + "\"}");
request.setEntity(reqEntity);

HttpResponse response = httpclient.execute(request);
HttpEntity entity = response.getEntity();

if (entity != null) {
System.out.println(EntityUtils.toString(entity));
}
} catch (Exception e) {
System.out.println(e.getMessage());
}
}

/**
* 获取缩略图
* 
* @param filePath
*            原始文件的url
* @param endPath
*            缩略图的url
*/
public static void getThumbnail(String filePath, String endPath) {
try {
URIBuilder builder = new URIBuilder(GET_THUMBNAIL);

builder.setParameter("width", "200");
builder.setParameter("height", "200");
builder.setParameter("smartCropping", "true");

URI uri = builder.build();
HttpPost request = new HttpPost(uri);
request.setHeader("Content-Type", "application/json");
request.setHeader("Ocp-Apim-Subscription-Key", KEY);

// Request body
StringEntity reqEntity = new StringEntity("{\"Url\": \"" + filePath + "\"}");
request.setEntity(reqEntity);

HttpResponse response = httpclient.execute(request);

HttpEntity entity = response.getEntity();

InputStream inputStream = entity.getContent();

// 将返回的图片输出
IOUtils.outputImg(inputStream, endPath);
} catch (Exception e) {
System.out.println(e.getMessage());
}
}

/**
* 获取特定领域模型列表（当前微软的API仅支持名人识别）
* 
* @param filePath
*/
public static JSONObject getListDomainSpecificModels() {
try {
URIBuilder builder = new URIBuilder(Domain_SPECIFIC);

URI uri = builder.build();
HttpGet request = new HttpGet(uri);
request.setHeader("Ocp-Apim-Subscription-Key", KEY);

HttpResponse response = httpclient.execute(request);
HttpEntity entity = response.getEntity();

String modelJson = EntityUtils.toString(entity);

if (entity != null) {
// System.out.println(modelJson);

return (JSONObject) JSONObject.parse(modelJson);
}
} catch (Exception e) {
System.out.println(e.getMessage());
}
return null;
}

/**
* 识别特定领域的内容(当前只支持名人识别),需要上一个接口getListDomainSpecificModels的支持
* 
* @param filePath
*/
public static void recognizeDomainSpecificContent(String filePath) {
// 首先从特定领域模型列表中选取需要的领域模型
JSONObject modelJson = getListDomainSpecificModels();

// 获取领域模型数组
JSONArray model = (JSONArray) modelJson.get("models");

// 遍历领域模型并找到与图片的领域相关信息
for (Object modelObj : model) {
JSONObject json = (JSONObject) modelObj;

String modelName = json.getString("name");

recognizeDomainSpecificContent(modelName, filePath);
}
}

/**
* 重载函数，用于识别给定领域的与图片相关的信息
* 
* @param modelName
* @param filePath
*/
public static void recognizeDomainSpecificContent(String modelName, String filePath) {
String modelUrl = Domain_SPECIFIC + "/" + modelName + "/analyze";

try {
URIBuilder builder = new URIBuilder(modelUrl);

URI uri = builder.build();
HttpPost request = new HttpPost(uri);
request.setHeader("Content-Type", "application/json");
request.setHeader("Ocp-Apim-Subscription-Key", KEY);

// Request body
StringEntity reqEntity = new StringEntity("{\"Url\":\"" + filePath + "\"}");
request.setEntity(reqEntity);

HttpResponse response = httpclient.execute(request);
HttpEntity entity = response.getEntity();

if (entity != null) {
System.out.println(EntityUtils.toString(entity));
}
} catch (Exception e) {
System.out.println(e.getMessage());
}
}

/**
* 获取图片中的文字信息
* 
* @param filePath
*/
public static JSONObject OCR(String filePath,String language) {
try {
URIBuilder builder = new URIBuilder(OCR);

/**
* 语言支持： unk (AutoDetect) -- 默认 zh-Hans (ChineseSimplified) zh-Hant
* (ChineseTraditional) cs (Czech) da (Danish) nl (Dutch) en
* (English) fi (Finnish) fr (French) de (German) el (Greek) hu
* (Hungarian) it (Italian) Ja (Japanese) ko (Korean) nb (Norwegian)
* pl (Polish) pt (Portuguese, ru (Russian) es (Spanish) sv
* (Swedish) tr (Turkish)
*/
builder.setParameter("language", language);
// 是否检测图像中的文本定位 true false
builder.setParameter("detectOrientation ", "true");

URI uri = builder.build();
HttpPost request = new HttpPost(uri);
// request.setHeader("Content-Type", "application/json");
request.setHeader("Content-Type", "application/octet-stream");
request.setHeader("Ocp-Apim-Subscription-Key", KEY);

// Request body
// StringEntity reqEntity = new StringEntity("{\"Url\":\"" +
// filePath + "\"}");
// request.setEntity(reqEntity);

File localFile = new File(filePath);

if (!localFile.exists()) {
System.out.println("图片不存在");

return null;
}

// 发送本地图片
// 以文件流方式访问时传递的是一个文件（最终会以位二进制进行请求）
FileEntity reqEntity = new FileEntity(localFile);
request.setEntity(reqEntity);

HttpResponse response = httpclient.execute(request);
HttpEntity entity = response.getEntity();

// HttpResponse response = httpclient.execute(request);
// HttpEntity entity = response.getEntity();

if (entity != null) {
String result = EntityUtils.toString(entity);

System.out.println();

return (JSONObject) JSONObject.parse(result);
}
} catch (Exception e) {
System.out.println(e.getMessage());
}

return null;
}

/**
* 获取图片的标签
* 
* @param filePath
*/
public static JSONObject getImgTag(String filePath) {
try {
URIBuilder builder = new URIBuilder(IMAGE_TAG);

URI uri = builder.build();
HttpPost request = new HttpPost(uri);
request.setHeader("Content-Type", "application/octet-stream");
request.setHeader("Ocp-Apim-Subscription-Key", KEY);

File localFile = new File(filePath);

if (!localFile.exists()) {
System.out.println("图片不存在");

return null;
}

// Request body

// StringEntity reqEntity = new StringEntity("{\"Url\":\"" + filePath + "\"}");

// request.setEntity(reqEntity);
FileEntity reqEntity = new FileEntity(localFile);
request.setEntity(reqEntity);

HttpResponse response = httpclient.execute(request);
HttpEntity entity = response.getEntity();

String result = EntityUtils.toString(entity);

return (JSONObject) JSONObject.parse(result);
} catch (Exception e) {
System.out.println(e.getMessage());
}

return null;
}

}

注意：调用时需要httpclient的jar包的支持