专栏首页微信公众号:Java团长Java基于百度API的图片文字识别

Java基于百度API的图片文字识别

具体文档:

http://ai.baidu.com/docs#/OCR-API/e1bd77f3

准备工作:

基于Java 1.8

版本控制:maven

使用之前需要获取对应的项目API_KEY,SECRET_KEY,这些参数在使用API的时候必须用到,用于生成access_token。

如何获取这些参数:在百度开发者中心申请一个“通用文字识别”项目,然后就可以获取到这些参数。

准备工作都完成了,现在开始进行图像识别了。

1. 准备pom文件

<!-- https://mvnrepository.com/artifact/com.alibaba/fastjson -->
<dependency>
 <groupId>com.alibaba</groupId>
 <artifactId>fastjson</artifactId>
 <version>1.2.46</version>
</dependency>
<!-- https://mvnrepository.com/artifact/org.apache.httpcomponents/httpclient -->
<dependency>
 <groupId>org.apache.httpcomponents</groupId>
 <artifactId>httpclient</artifactId>
 <version>4.5.5</version>
</dependency>

2. 获取access_token

package com.wsk.netty.check;

import org.json.JSONObject;

import java.io.BufferedReader;
import java.io.InputStreamReader;
import java.net.HttpURLConnection;
import java.net.URL;
import java.util.List;
import java.util.Map;
/**
* 获取token类
*
* @Author : WuShukai
* @Date :2018/2/12 10:04
*/

public class AuthService {

   /**
    * 获取权限token
    * @return 返回示例:
    * {
    * "access_token": "24.460da4889caad24cccdb1fea17221975.2592000.1491995545.282335-1234567",
     * "expires_in": 2592000
    * }
    */
   public static String getAuth() {
       // 官网获取的 API Key 更新为你注册的
       String clientId = "**";
       // 官网获取的 Secret Key 更新为你注册的
       String clientSecret = "**";
       return getAuth(clientId, clientSecret);
   }

   /**
    * 获取API访问token
    * 该token有一定的有效期,需要自行管理,当失效时需重新获取.
    * @param ak - 百度云官网获取的 API Key
    * @param sk - 百度云官网获取的 Securet Key
    * @return assess_token 示例:
    * "24.460da4889caad24cccdb1fea17221975.2592000.1491995545.282335-1234567"
    */
   private 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()));
            StringBuilder result = new StringBuilder();
           String line;
           while ((line = in.readLine()) != null) {
               result.append(line);
           }
           /**
            * 返回结果示例
            */
           System.err.println("result:" + result);
           JSONObject jsonObject = new JSONObject(result.toString());
           return jsonObject.getString("access_token");
       } catch (Exception e) {
           System.err.printf("获取token失败!");
           e.printStackTrace(System.err);
       }
       return null;
   }

   public static void main(String[] args) {
       getAuth();
   }

}

3. 编写将图片转化成base64后再转化成urlencode的工具类

package com.wsk.netty.check;

import sun.misc.BASE64Encoder;

import java.io.FileInputStream;
import java.io.IOException;
import java.io.InputStream;
import java.net.URLEncoder;

/**
* 图片转化base64后再UrlEncode结果
* @Author : WuShukai
* @Date :2018/2/12 10:43
*/

public class BaseImg64 {
   /**
    * 将一张本地图片转化成Base64字符串
    * @param imgPath 本地图片地址
    * @return 图片转化base64后再UrlEncode结果
    */
   public static String getImageStrFromPath(String imgPath) {
       InputStream in;
       byte[] data = null;
       // 读取图片字节数组
       try {
           in = new FileInputStream(imgPath);
           data = new byte[in.available()];
           in.read(data);
           in.close();
       } catch (IOException e) {
           e.printStackTrace();
       }
       // 对字节数组Base64编码
       BASE64Encoder encoder = new BASE64Encoder();
       // 返回Base64编码过再URLEncode的字节数组字符串
       return URLEncoder.encode(encoder.encode(data));
   }
}

4. 编写调用百度API接口的方法,获取识别结果

package com.wsk.netty.check;

import org.apache.http.HttpResponse;
import org.apache.http.client.HttpClient;
import org.apache.http.client.methods.HttpPost;
import org.apache.http.entity.StringEntity;
import org.apache.http.impl.client.DefaultHttpClient;
import org.apache.http.util.EntityUtils;

import java.io.File;
import java.io.IOException;
import java.net.URI;
import java.net.URISyntaxException;

/**
* 图像文字识别
*
* @Author : WuShukai
* @Date :2018/2/12 10:25
*/

public class Check {
   private static final String POST_URL = "https://aip.baidubce.com/rest/2.0/ocr/v1/general_basic?access_token=" + AuthService.getAuth();

   /**
    * 识别本地图片的文字
    *
    * @param path 本地图片地址
    * @return 识别结果,为json格式
    * @throws URISyntaxException URI打开异常
    * @throws IOException        io流异常
    */
   public static String checkFile(String path) throws URISyntaxException, IOException {
       File file = new File(path);
       if (!file.exists()) {
           throw new NullPointerException("图片不存在");
       }
       String image = BaseImg64.getImageStrFromPath(path);
       String param = "image=" + image;
       return post(param);
   }

   /**
    * @param url 图片url
    * @return 识别结果,为json格式
    */
   public static String checkUrl(String url) throws IOException, URISyntaxException {
       String param = "url=" + url;
       return post(param);
   }

   /**
    * 通过传递参数:url和image进行文字识别
    *
    * @param param 区分是url还是image识别
    * @return 识别结果
    * @throws URISyntaxException URI打开异常
    * @throws IOException        IO流异常
    */
   private static String post(String param) throws URISyntaxException, IOException {
       //开始搭建post请求
       HttpClient httpClient = new DefaultHttpClient();
       HttpPost post = new HttpPost();
       URI url = new URI(POST_URL);
       post.setURI(url);
       //设置请求头,请求头必须为application/x-www-form-urlencoded,因为是传递一个很长的字符串,不能分段发送
        post.setHeader("Content-Type", "application/x-www-form-urlencoded");
       StringEntity entity = new StringEntity(param);
       post.setEntity(entity);
       HttpResponse response = httpClient.execute(post);
       System.out.println(response.toString());
       if (response.getStatusLine().getStatusCode() == 200) {
           String str;
           try {
               /*读取服务器返回过来的json字符串数据*/
               str = EntityUtils.toString(response.getEntity());
               System.out.println(str);
               return str;
           } catch (Exception e) {
               e.printStackTrace();
               return null;
           }
       }
       return null;
   }

   public static void main(String[] args) {
       String path = "E:\\find.png";
       try {
           long now = System.currentTimeMillis();
           checkFile(path);
           checkUrl("https://gss3.bdstatic.com/-Po3dSag_xI4khGkpoWK1HF6hhy/baike/c0%3Dbaike80%2C5%2C5%2C80%2C26/sign=08c05c0e8444ebf8797c6c6db890bc4f/fc1f4134970a304e46bfc5f7d2c8a786c9175c19.jpg");
            System.out.println("耗时:" + (System.currentTimeMillis() - now) / 1000 + "s");
        } catch (URISyntaxException | IOException e) {
           e.printStackTrace();
       }
   }
}

5. 识别结果(仅测试本地图片识别)

中文

结果:

结论:

这里是使用了Postman进行测试的,用IDEA控制台的话,返回的json不易读。 从这里可以看出,耗时是1s,虽然识别率高,但是结果还是有那么的一些差距,例如识别结果的第五列,只返回了“我是逊尼”,而原图片的很大串没有识别出来。

英文:

结果:

结论:

单识别英文的图片,效果还是比较满意的,耗时短,精准率高。

中英文结合:

结果:

结论:

结果也是比较满意的。

本文分享自微信公众号 - Java团长(javatuanzhang),作者:wsk1103

原文出处及转载信息见文内详细说明,如有侵权,请联系 yunjia_community@tencent.com 删除。

原始发表时间:2018-09-26

本文参与腾讯云自媒体分享计划,欢迎正在阅读的你也加入,一起分享。

我来说两句

0 条评论
登录 后参与评论

相关文章

  • Java POI 导出EXCEL经典实现

    在web开发中,有一个经典的功能,就是数据的导入导出。特别是数据的导出,在生产管理或者财务系统中用的非常普遍,因为这些系统经常要做一些报表打印的工作。而数据导出...

    Java团长
  • SpringBoot+Shiro+Redis共享Session入门小栗子

    在单机版的Springboot+Shiro的基础上,这次实现共享Session。

    Java团长
  • 这篇带你深入理解SpringBoot中的自动装配(好文精读)

    SpringBoot的自动装配是拆箱即用的基础,也是微服务化的前提。其实它并不那么神秘,我在这之前已经写过最基本的实现了,大家可以参考这篇文章。这次主要的议题是...

    Java团长
  • Java基于百度API的图片文字识别(支持中文,英文和中英文混合)

    具体文档:http://ai.baidu.com/docs#/OCR-API/e1bd77f3

    好好学java
  • xml与数据库中数据的导入导出

    这是我一个晚上做出来的,因为要去做其他的项目,所以只实现了对特定数据库的xml操作,不过我觉得这是学习xml挺不错的参考代码和文档

    SmileNicky
  • JSP上传文件与导出Excel表

    我们可以通过Apache的fileupload组件来实现jsp上传文件,这样就不需要自己去写具体的实现了,首先需要配置如下依赖:

    端碗吹水
  • Java爬虫

    jsoup-1.7.3.jar 个人认为爬虫的实现机制: 获取Docume对象—>获取节点—>输出或者持久化

    用户1518699
  • SpringBoot项目:RedisTemplate实现轻量级消息队列(含代码)

    1、excel文件读写--阿里easyexcel sdk 2、文件上传、下载--腾讯云对象存储 3、远程服务调用--restTemplate 4、生产者、消费者...

    搜云库技术团队
  • 【Java多线程】写入同一文件,自定义线程池与线程回收利用2 顶

    起初为了方便快捷,只为实现功能,写了很多垃圾的代码. 造成性能不高,可读性,可维护性不强。

    linapex
  • 性能测试如何减少本机误差

    在接口测试过程中,某个线程连续两次请求中间都会消耗的时间,线程在收到响应进行第二次请求的时候会消耗一些时间,比如进行结果验证,日志存储,或者进行数据统计等等。

    FunTester

扫码关注云+社区

领取腾讯云代金券