百度云识别图文

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
90
91
92
93
94
package com.example.demo.baidu;


import java.io.BufferedReader;
import java.io.InputStreamReader;
import java.net.HttpURLConnection;
import java.net.URL;
import java.net.URLEncoder;
import java.util.List;
import java.util.Map;

import org.json.JSONObject;

import com.example.demo.baidu.FileUtil;
import com.example.demo.baidu.HttpUtil;
import com.example.demo.baidu.Base64Util;

/**
 * 百度云识别图文
 * @author 闫钰琦
 *
 */

public class baidu1 {

	
	public static void main(String[] args) {
		
		String otherHost = "https://aip.baidubce.com/rest/2.0/ocr/v1/general";
	        // 本地图片路径
	        String filePath = "D:/one.jpeg";
	        try {
	            byte[] imgData = FileUtil.readFileByBytes(filePath);
	            String imgStr = Base64Util.encode(imgData);
	            String params = URLEncoder.encode("image", "UTF-8") + "=" + URLEncoder.encode(imgStr, "UTF-8");
	            /**
	             * 线上环境access_token有过期时间， 客户端可自行缓存，过期后重新获取。
	             */
	            String accessToken = getAuth("写入自己的token", "在百度云上注册可直接查看");
	            String result = HttpUtil.post(otherHost, accessToken, params);
	            System.out.println(result);
	        } catch (Exception e) {
	            e.printStackTrace();
	        }
		
		
	}


	
	 public static String getAuth(String ak, String sk) {
	        // 获取token地址
	        String authHost = "https://aip.baidubce.com/oauth/2.0/token?";
	        String getAccessTokenUrl = authHost
	                // 1. grant_type为固定参数
	                + "grant_type=client_credentials"
	                // 2. 官网获取的 API Key
	                + "&client_id=" + ak
	                // 3. 官网获取的 Secret Key
	                + "&client_secret=" + sk;
	        try {
	            URL realUrl = new URL(getAccessTokenUrl);
	            // 打开和URL之间的连接
	            HttpURLConnection connection = (HttpURLConnection) realUrl.openConnection();
	            connection.setRequestMethod("GET");
	            connection.connect();
	            // 获取所有响应头字段
	            Map<String, List<String>> map = connection.getHeaderFields();
	            // 遍历所有的响应头字段
	            for (String key : map.keySet()) {
	                System.err.println(key + "--->" + map.get(key));
	            }
	            // 定义 BufferedReader输入流来读取URL的响应
	            BufferedReader in = new BufferedReader(new InputStreamReader(connection.getInputStream()));
	            String result = "";
	            String line;
	            while ((line = in.readLine()) != null) {
	                result += line;
	            }
	            /**
	             * 返回结果示例
	             */
	            System.err.println("result:" + result);
	            JSONObject jsonObject = new JSONObject(result);
	            String access_token = jsonObject.getString("access_token");
	            return access_token;
	        } catch (Exception e) {
	            System.err.printf("获取token失败！");
	            e.printStackTrace(System.err);
	        }
	        return null;
	    }
	
}

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
package com.example.demo.baidu;

/**
* Base64 工具类
*/
public class Base64Util {
   private static final char last2byte = (char) Integer.parseInt("00000011", 2);
   private static final char last4byte = (char) Integer.parseInt("00001111", 2);
   private static final char last6byte = (char) Integer.parseInt("00111111", 2);
   private static final char lead6byte = (char) Integer.parseInt("11111100", 2);
   private static final char lead4byte = (char) Integer.parseInt("11110000", 2);
   private static final char lead2byte = (char) Integer.parseInt("11000000", 2);
   private static final char[] encodeTable = new char[]{'A', 'B', 'C', 'D', 'E', 'F', 'G', 'H', 'I', 'J', 'K', 'L', 'M', 'N', 'O', 'P', 'Q', 'R', 'S', 'T', 'U', 'V', 'W', 'X', 'Y', 'Z', 'a', 'b', 'c', 'd', 'e', 'f', 'g', 'h', 'i', 'j', 'k', 'l', 'm', 'n', 'o', 'p', 'q', 'r', 's', 't', 'u', 'v', 'w', 'x', 'y', 'z', '0', '1', '2', '3', '4', '5', '6', '7', '8', '9', '+', '/'};

   public Base64Util() {
   }

   public static String encode(byte[] from) {
       StringBuilder to = new StringBuilder((int) ((double) from.length * 1.34D) + 3);
       int num = 0;
       char currentByte = 0;

       int i;
       for (i = 0; i < from.length; ++i) {
           for (num %= 8; num < 8; num += 6) {
               switch (num) {
                   case 0:
                       currentByte = (char) (from[i] & lead6byte);
                       currentByte = (char) (currentByte >>> 2);
                   case 1:
                   case 3:
                   case 5:
                   default:
                       break;
                   case 2:
                       currentByte = (char) (from[i] & last6byte);
                       break;
                   case 4:
                       currentByte = (char) (from[i] & last4byte);
                       currentByte = (char) (currentByte << 2);
                       if (i + 1 < from.length) {
                           currentByte = (char) (currentByte | (from[i + 1] & lead2byte) >>> 6);
                       }
                       break;
                   case 6:
                       currentByte = (char) (from[i] & last2byte);
                       currentByte = (char) (currentByte << 4);
                       if (i + 1 < from.length) {
                           currentByte = (char) (currentByte | (from[i + 1] & lead4byte) >>> 4);
                       }
               }

               to.append(encodeTable[currentByte]);
           }
       }

       if (to.length() % 4 != 0) {
           for (i = 4 - to.length() % 4; i > 0; --i) {
               to.append("=");
           }
       }

       return to.toString();
   }
}

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
package com.example.demo.baidu;
import java.io.*;

/**
 * 文件读取工具类
 */
public class FileUtil {
 
    /**
     * 读取文件内容，作为字符串返回
     */
    public static String readFileAsString(String filePath) throws IOException {
        File file = new File(filePath);
        if (!file.exists()) {
            throw new FileNotFoundException(filePath);
        } 
 
        if (file.length() > 1024 * 1024 * 1024) {
            throw new IOException("File is too large");
        } 
 
        StringBuilder sb = new StringBuilder((int) (file.length()));
        // 创建字节输入流  
        FileInputStream fis = new FileInputStream(filePath);  
        // 创建一个长度为10240的Buffer
        byte[] bbuf = new byte[10240];  
        // 用于保存实际读取的字节数  
        int hasRead = 0;  
        while ( (hasRead = fis.read(bbuf)) > 0 ) {  
            sb.append(new String(bbuf, 0, hasRead));  
        }  
        fis.close();  
        return sb.toString();
    }
 
    /**
     * 根据文件路径读取byte[] 数组
     */
    public static byte[] readFileByBytes(String filePath) throws IOException {
        File file = new File(filePath);
        if (!file.exists()) {
            throw new FileNotFoundException(filePath);
        } else {
            ByteArrayOutputStream bos = new ByteArrayOutputStream((int) file.length());
            BufferedInputStream in = null;
 
            try {
                in = new BufferedInputStream(new FileInputStream(file));
                short bufSize = 1024;
                byte[] buffer = new byte[bufSize];
                int len1;
                while (-1 != (len1 = in.read(buffer, 0, bufSize))) {
                    bos.write(buffer, 0, len1);
                }
 
                byte[] var7 = bos.toByteArray();
                return var7;
            } finally {
                try {
                    if (in != null) {
                        in.close();
                    }
                } catch (IOException var14) {
                    var14.printStackTrace();
                }
 
                bos.close();
            }
        }
    }
}

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
package com.example.demo.baidu;

import java.io.BufferedReader;
import java.io.DataOutputStream;
import java.io.InputStreamReader;
import java.net.HttpURLConnection;
import java.net.URL;
import java.util.List;
import java.util.Map;

/**
* http 工具类
*/
public class HttpUtil {

   public static String post(String requestUrl, String accessToken, String params)
           throws Exception {
       String contentType = "application/x-www-form-urlencoded";
       return HttpUtil.post(requestUrl, accessToken, contentType, params);
   }

   public static String post(String requestUrl, String accessToken, String contentType, String params)
           throws Exception {
       String encoding = "UTF-8";
       if (requestUrl.contains("nlp")) {
           encoding = "GBK";
       }
       return HttpUtil.post(requestUrl, accessToken, contentType, params, encoding);
   }

   public static String post(String requestUrl, String accessToken, String contentType, String params, String encoding)
           throws Exception {
       String url = requestUrl + "?access_token=" + accessToken;
       return HttpUtil.postGeneralUrl(url, contentType, params, encoding);
   }

   public static String postGeneralUrl(String generalUrl, String contentType, String params, String encoding)
           throws Exception {
       URL url = new URL(generalUrl);
       // 打开和URL之间的连接
       HttpURLConnection connection = (HttpURLConnection) url.openConnection();
       connection.setRequestMethod("POST");
       // 设置通用的请求属性
       connection.setRequestProperty("Content-Type", contentType);
       connection.setRequestProperty("Connection", "Keep-Alive");
       connection.setUseCaches(false);
       connection.setDoOutput(true);
       connection.setDoInput(true);

       // 得到请求的输出流对象
       DataOutputStream out = new DataOutputStream(connection.getOutputStream());
       out.write(params.getBytes(encoding));
       out.flush();
       out.close();

       // 建立实际的连接
       connection.connect();
       // 获取所有响应头字段
       Map<String, List<String>> headers = connection.getHeaderFields();
       // 遍历所有的响应头字段
       for (String key : headers.keySet()) {
           System.err.println(key + "--->" + headers.get(key));
       }
       // 定义 BufferedReader输入流来读取URL的响应
       BufferedReader in = null;
       in = new BufferedReader(
               new InputStreamReader(connection.getInputStream(), encoding));
       String result = "";
       String getLine;
       while ((getLine = in.readLine()) != null) {
           result += getLine;
       }
       in.close();
       System.err.println("result:" + result);
       return result;
   }
}