package org.springblade.modules.test4j;
|
|
import io.swagger.annotations.Api;
|
import lombok.AllArgsConstructor;
|
import net.sourceforge.tess4j.ITesseract;
|
import net.sourceforge.tess4j.Tesseract;
|
import net.sourceforge.tess4j.TesseractException;
|
import org.springblade.core.tool.api.R;
|
import org.springblade.modules.test4j.util.Test4jUtil;
|
import org.springframework.web.bind.annotation.PostMapping;
|
import org.springframework.web.bind.annotation.RequestMapping;
|
import org.springframework.web.bind.annotation.RequestParam;
|
import org.springframework.web.bind.annotation.RestController;
|
import org.springframework.web.multipart.MultipartFile;
|
|
import javax.imageio.ImageIO;
|
import java.awt.*;
|
import java.awt.image.BufferedImage;
|
import java.io.IOException;
|
import java.util.HashMap;
|
import java.util.Map;
|
|
@RestController
|
@AllArgsConstructor
|
@RequestMapping("/test4j")
|
@Api(value = "图片智能识别", tags = "图片智能识别")
|
public class Test4jController {
|
|
@PostMapping("/read-id")
|
public R readText(@RequestParam("file") MultipartFile file) {
|
try {
|
BufferedImage grayscaleImage = grayscale(ImageIO.read(file.getInputStream()));
|
String text = recognizeText(grayscaleImage);
|
return R.data(text);
|
} catch (IOException e) {
|
e.printStackTrace();
|
return null;
|
}
|
}
|
|
public BufferedImage grayscale(BufferedImage image) {
|
int width = image.getWidth();
|
int height = image.getHeight();
|
BufferedImage result = new BufferedImage(width, height, BufferedImage.TYPE_BYTE_GRAY);
|
Graphics g = result.getGraphics();
|
g.drawImage(image, 0, 0, null);
|
g.dispose();
|
return result;
|
}
|
|
public String recognizeText(BufferedImage image) {
|
ITesseract tesseract = new Tesseract();
|
// 设置Tesseract的路径
|
tesseract.setDatapath("F:\\test4jdata");
|
// 设置为中文简体
|
tesseract.setLanguage("chi_sim");
|
try {
|
return tesseract.doOCR(image);
|
} catch (TesseractException e) {
|
e.printStackTrace();
|
return null;
|
}
|
}
|
}
|
