调用微软小冰API,实现批量人脸颜值打分

源码:


import net.sf.json.JSONObject;
import sun.misc.BASE64Encoder;
import org.apache.http.HttpResponse;
import org.apache.http.NameValuePair;
import org.apache.http.client.CookieStore;
import org.apache.http.client.entity.UrlEncodedFormEntity;
import org.apache.http.client.methods.HttpPost;
import org.apache.http.impl.client.DefaultHttpClient;
import org.apache.http.message.BasicNameValuePair;
import org.apache.http.util.EntityUtils;
import java.io.*;
import java.net.HttpURLConnection;
import java.net.URL;
import java.util.ArrayList;
import java.util.Calendar;
import java.util.List;
import java.util.Map;
import java.util.regex.Matcher;
import java.util.regex.Pattern;

public class AnalyseImage {
	private static final String upload_picBase64_api = "http://kan.msxiaobing.com/APi/Image/UploadBase64";
	private static final String ice_api = "http://kan.msxiaobing.com/Api/ImageAnalyze/Process?service=yanzhi&tid=8b9a88049f9c4f26b4da60afc9d70ef4";
	private static final String ice_page = "http://kan.msxiaobing.com/ImageGame/Portal?task=yanzhi";

	private static String getUploadPicResult(String imgdataBase64) {
		StringBuffer sb = new StringBuffer();
		try {
			URL realUrl = new URL(upload_picBase64_api);
			HttpURLConnection conn = (HttpURLConnection) realUrl.openConnection();
			conn.setDoOutput(true);
			conn.setDoInput(true);
			conn.setUseCaches(false);
			conn.setRequestMethod("POST");

			conn.setRequestProperty("Charset", "UTF-8");
			conn.setRequestProperty("Content-Type", "application/x-www-form-urlencoded");
			conn.setRequestProperty("Connection", "Keep-Alive");// 维持长连接

			DataOutputStream dataOutputStream = new DataOutputStream(conn.getOutputStream());
			dataOutputStream.writeBytes(imgdataBase64);
			dataOutputStream.flush();
			dataOutputStream.close();

			String readLine = new String();
			BufferedReader responseReader = new BufferedReader(new InputStreamReader(conn.getInputStream(), "UTF-8"));
			while ((readLine = responseReader.readLine()) != null) {
				sb.append(readLine).append("\n");
			}
			responseReader.close();
			System.out.println(sb.toString());

		} catch (Exception e) {
			e.printStackTrace();
		} finally {
			return sb.toString();
		}
	}

	/**
	 * 分析照片
	 */
	public static String analyzeImage(String jsonResultPic) {
		DefaultHttpClient httpClient = new DefaultHttpClient();
		DefaultHttpClient httpClient2 = new DefaultHttpClient();
		HttpPost httpPost1 = new HttpPost(ice_api);
		HttpPost httpPost2 = new HttpPost(ice_page);

		JSONObject jasonObject = JSONObject.fromObject(jsonResultPic);
		Map<String, String> contentImgUrl = (Map) jasonObject;
		String imgUrl = contentImgUrl.get("Host") + contentImgUrl.get("Url");
		System.out.print(jsonResultPic);
		List<NameValuePair> form = new ArrayList<NameValuePair>();

		form.add(new BasicNameValuePair("MsgId", String.valueOf(System.currentTimeMillis()) + "063"));
		form.add(new BasicNameValuePair("CreateTime", String.valueOf(Calendar.getInstance().getTimeInMillis())));
		form.add(new BasicNameValuePair("Content[imageUrl]", imgUrl));

		try {
			httpPost1.addHeader("Referer",
					"http://kan.msxiaobing.com/V3/Portal?task=yanzhi&ftid=91ac082228fb48739f12c66ee3a3fee0");
			httpPost1.setEntity(new UrlEncodedFormEntity(form, "UTF-8"));

			HttpResponse pageResponse = httpClient.execute(httpPost2);
			CookieStore cookieStore = httpClient.getCookieStore();
			httpClient2.setCookieStore(cookieStore);
			HttpResponse response = httpClient2.execute(httpPost1);
			return EntityUtils.toString(response.getEntity());
		} catch (Exception e) {
			e.printStackTrace();
		}
		return null;

	}

	/**
	 * 获取颜值分数
	 * 
	 * @param jsonResultPic
	 * @return
	 */
	public static int findScoreFromString(String jsonResultPic) {
		Pattern pattern = Pattern.compile("\\d+[.]\\d+");
		Matcher m = pattern.matcher(jsonResultPic);
		if (m.find()) {
			System.out.println("analyzeResult=" + m.group());
			double temp = Double.valueOf(m.group());
			int score = (int) (temp * 10);
			return score;
		} else {
			return 0;
		}
	}

	/**
	 * 将文件转成base64 字符串
	 */
	public static String encodeBase64File(File file) throws Exception {
		FileInputStream inputFile = new FileInputStream(file);
		byte[] buffer = new byte[(int) file.length()];
		inputFile.read(buffer);
		inputFile.close();
		return new BASE64Encoder().encode(buffer);
	}

	public static void main(String[] args) throws Exception {
		//图片目录
		File dir = new File("C:\\Users\\chw93\\Desktop\\face");
		File[] listFiles = dir.listFiles();
		
		// 分析结果记录
		File result = new File("C:\\Users\\chw93\\Desktop\\result.txt");
		FileWriter fw = new FileWriter(result, true);
		PrintWriter pw = new PrintWriter(fw);
		for (File f : listFiles) {
			// 读取文件图片为base64编码格式
			String imgDataBase64 = FileToBase64.encodeBase64File(f);
			// base64上传到微软服务器
			String jsonResultPic = getUploadPicResult(imgDataBase64);
			String analyzeResult = analyzeImage(jsonResultPic);
			int faceAuthScore = findScoreFromString(analyzeResult);
			System.out.println("得分:" + faceAuthScore);
	     pw.println(f.getName()+",得分:"+faceAuthScore);
	 	 pw.flush();
		}

		fw.flush();pw.close();fw.close();
	}

}

 

依赖jar包:

<!--base64解码/加密 -->
		<dependency>
			<groupId>commons-codec</groupId>
			<artifactId>commons-codec</artifactId>
			<version>1.10</version><!--$NO-MVN-MAN-VER$-->
		</dependency>
		<!--引入html测试需要的包 -->
		<dependency>
			<groupId>org.springframework.boot</groupId>
			<artifactId>spring-boot-starter-thymeleaf</artifactId>
		</dependency>
		<!--http请求的包-->
		<dependency>
			<groupId>org.apache.httpcomponents</groupId>
			<artifactId>httpclient</artifactId>
			<version>4.3.3</version><!--$NO-MVN-MAN-VER$ -->
		</dependency>
		<!-- json start -->
		<dependency>
			<groupId>net.sf.json-lib</groupId>
			<artifactId>json-lib</artifactId>
			<version>2.4</version>
			<classifier>jdk15</classifier>
		</dependency>
		<!-- json end -->

 

附图:


 

评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值