首页 > 代码库 > Java调用OCR进行图片识别

Java调用OCR进行图片识别

使用Java语言,通过Tesseract-OCR对图片进行识别。

1.Tesseract-OCR

下载windows版本并安装。

2.程序如下:

a.ImageIOHelper类

 

[java] view plaincopy技术分享技术分享
 
  1. package OCR;  
  2.   
  3. import java.awt.image.BufferedImage;  
  4. import java.io.File;  
  5. import java.io.IOException;  
  6. import java.util.Iterator;  
  7. import java.util.Locale;  
  8.   
  9. import javax.imageio.IIOImage;  
  10. import javax.imageio.ImageIO;  
  11. import javax.imageio.ImageReader;  
  12. import javax.imageio.ImageWriteParam;  
  13. import javax.imageio.ImageWriter;  
  14. import javax.imageio.metadata.IIOMetadata;  
  15. import javax.imageio.stream.ImageInputStream;  
  16. import javax.imageio.stream.ImageOutputStream;  
  17.   
  18. import com.sun.media.imageio.plugins.tiff.TIFFImageWriteParam;  
  19.   
  20. public class ImageIOHelper {  
  21.     /** 
  22.      * 图片文件转换为tif格式 
  23.      * @param imageFile 文件路径 
  24.      * @param imageFormat 文件扩展名 
  25.      * @return 
  26.      */  
  27.     public static File createImage(File imageFile, String imageFormat) {  
  28.         File tempFile = null;  
  29.         try {  
  30.             Iterator<ImageReader> readers = ImageIO.getImageReadersByFormatName(imageFormat);  
  31.             ImageReader reader = readers.next();  
  32.           
  33.             ImageInputStream iis = ImageIO.createImageInputStream(imageFile);  
  34.             reader.setInput(iis);  
  35.             //Read the stream metadata  
  36.             IIOMetadata streamMetadata = reader.getStreamMetadata();  
  37.               
  38.             //Set up the writeParam  
  39.             TIFFImageWriteParam tiffWriteParam = new TIFFImageWriteParam(Locale.CHINESE);  
  40.             tiffWriteParam.setCompressionMode(ImageWriteParam.MODE_DISABLED);  
  41.               
  42.             //Get tif writer and set output to file  
  43.             Iterator<ImageWriter> writers = ImageIO.getImageWritersByFormatName("tiff");  
  44.             ImageWriter writer = writers.next();  
  45.               
  46.             BufferedImage bi = reader.read(0);  
  47.             IIOImage image = new IIOImage(bi,null,reader.getImageMetadata(0));  
  48.             tempFile = tempImageFile(imageFile);  
  49.             ImageOutputStream ios = ImageIO.createImageOutputStream(tempFile);  
  50.             writer.setOutput(ios);  
  51.             writer.write(streamMetadata, image, tiffWriteParam);  
  52.             ios.close();  
  53.               
  54.             writer.dispose();  
  55.             reader.dispose();  
  56.               
  57.         } catch (IOException e) {  
  58.             e.printStackTrace();  
  59.         }  
  60.         return tempFile;  
  61.     }  
  62.   
  63.     private static File tempImageFile(File imageFile) {  
  64.         String path = imageFile.getPath();  
  65.         StringBuffer strB = new StringBuffer(path);  
  66.         strB.insert(path.lastIndexOf(‘.‘),0);  
  67.         return new File(strB.toString().replaceFirst("(?<=//.)(//w+)$", "tif"));  
  68.     }  
  69.   
  70. }  

b.OCR核心类

 

 

[java] view plaincopy技术分享技术分享
 
  1. package OCR;  
  2.   
  3. import java.io.BufferedReader;  
  4. import java.io.File;  
  5. import java.io.FileInputStream;  
  6. import java.io.InputStreamReader;  
  7. import java.util.ArrayList;  
  8. import java.util.List;  
  9.   
  10. import org.jdesktop.swingx.util.OS;  
  11.   
  12. public class OCR {  
  13.     private final String LANG_OPTION = "-l";  //英文字母小写l,并非数字1  
  14.     private final String EOL = System.getProperty("line.separator");  
  15.     private String tessPath = "C://Program Files//Tesseract-OCR";  
  16.     //private String tessPath = new File("tesseract").getAbsolutePath();  
  17.       
  18.     public String recognizeText(File imageFile,String imageFormat)throws Exception{  
  19.         File tempImage = ImageIOHelper.createImage(imageFile,imageFormat);  
  20.         File outputFile = new File(imageFile.getParentFile(),"output");  
  21.         StringBuffer strB = new StringBuffer();  
  22.         List<String> cmd = new ArrayList<String>();  
  23.         if(OS.isWindowsXP()){  
  24.             cmd.add(tessPath+"//tesseract");  
  25.         }else if(OS.isLinux()){  
  26.             cmd.add("tesseract");  
  27.         }else{  
  28.             cmd.add(tessPath+"//tesseract");  
  29.         }  
  30.         cmd.add("");  
  31.         cmd.add(outputFile.getName());  
  32.         //cmd.add(LANG_OPTION);  
  33.         //cmd.add("chi_sim");  
  34.         //cmd.add("eng");  
  35.           
  36.         ProcessBuilder pb = new ProcessBuilder();  
  37.         pb.directory(imageFile.getParentFile());  
  38.           
  39.         cmd.set(1, tempImage.getName());  
  40.         pb.command(cmd);  
  41.         pb.redirectErrorStream(true);  
  42.           
  43.         Process process = pb.start();  
  44.         //tesseract.exe 1.jpg 1 -l chi_sim  
  45.         int w = process.waitFor();  
  46.           
  47.         //删除临时正在工作文件  
  48.         tempImage.delete();  
  49.           
  50.         if(w==0){  
  51.             BufferedReader in = new BufferedReader(new InputStreamReader(new FileInputStream(outputFile.getAbsolutePath()+".txt"),"UTF-8"));  
  52.               
  53.             String str;  
  54.             while((str = in.readLine())!=null){  
  55.                 strB.append(str).append(EOL);  
  56.             }  
  57.             in.close();  
  58.         }else{  
  59.             String msg;  
  60.             switch(w){  
  61.                 case 1:  
  62.                     msg = "Errors accessing files.There may be spaces in your image‘s filename.";  
  63.                     break;  
  64.                 case 29:  
  65.                     msg = "Cannot recongnize the image or its selected region.";  
  66.                     break;  
  67.                 case 31:  
  68.                     msg = "Unsupported image format.";  
  69.                     break;  
  70.                 default:  
  71.                     msg = "Errors occurred.";  
  72.             }  
  73.             tempImage.delete();  
  74.             //throw new RuntimeException(msg);  
  75.         }  
  76.         new File(outputFile.getAbsolutePath()+".txt").delete();  
  77.         return strB.toString();  
  78.     }  
  79. }  

c.main

 

 

[java] view plaincopy技术分享技术分享
 
  1. package OCR;  
  2. import java.io.File;  
  3. import java.io.IOException;  
  4.   
  5. public class TestOcr {  
  6.   
  7.     /** 
  8.      * @param args 
  9.      */  
  10.     public static void main(String[] args) {  
  11.         //输入图片地址  
  12.         String path = "d://test//test.bmp";     
  13.         try {     
  14.             String valCode = new OCR().recognizeText(new File(path), "bmp");     
  15.             System.out.println(valCode);     
  16.         } catch (IOException e) {     
  17.             e.printStackTrace();     
  18.         } catch (Exception e) {  
  19.             e.printStackTrace();  
  20.         }      
  21.     }  
  22.   
  23. }  

Java调用OCR进行图片识别