SpringBoot結合Tess4J實現(xiàn)拍圖識字的示例代碼

更新時間：2024年06月05日 10:51:23 作者：Swittee丶Henry

圖片中的文字提取已經(jīng)越來越多地應用于數(shù)據(jù)輸入和自動化處理過程,本文主要介紹了SpringBoot結合Tess4J實現(xiàn)拍圖識字的示例代碼,具有一定的參考價值,感興趣的可以了解一下

背景

隨著信息技術的不斷進步，圖片中的文字提取已經(jīng)越來越多地應用于數(shù)據(jù)輸入和自動化處理過程。Tess4J，作為Tesseract OCR引擎的Java JNA封裝，提供了一個能力強大的接口來實現(xiàn)這一功能。

在Spring Boot中整合Tess4J，我們可以快速地在Java應用中優(yōu)雅地實現(xiàn)文字識別。本指南將手把手教你在Spring Boot項目中實現(xiàn)這一功能。

第1部分：環(huán)境搭建

在開始之前，請確保你有以下環(huán)境配置：

JDK 1.8或更高版本
Maven
最新版的Spring Boot
Tess4J版本4.x或更高

第2部分：添加依賴

在你的pom.xml中加入以下依賴，以便于使用Tess4J：

<dependencies>
    <dependency>
        <groupId>net.sourceforge.tess4j</groupId>
        <artifactId>tess4j</artifactId>
        <version>4.5.4</version>
    </dependency>
    <!-- 其他依賴 -->
</dependencies>

確保以上版本是最新的，或者是適配當前開發(fā)環(huán)境的版本。

添加Tessdata語言庫

github下：

https://gitcode.com/tesseract-ocr/tessdata/tree/main

第3部分：創(chuàng)建OCR服務類

@Service
public class OcrService {

    public String recognizeText(File imageFile) throws TesseractException {
        Tesseract tesseract = new Tesseract();
        
        // 設定訓練文件的位置（如果是標準英文識別，此步可省略）
        tesseract.setDatapath("你的tessdata各語言集合包地址");
        tesseract.setLanguage("chi_sim");
        return tesseract.doOCR(imageFile);
    }

    public String recognizeTextFromUrl(String imageUrl) throws Exception {
        URL url = new URL(imageUrl);
        InputStream in = url.openStream();
        Files.copy(in, Paths.get("downloaded.jpg"), StandardCopyOption.REPLACE_EXISTING);

        File imageFile = new File("downloaded.jpg");
        return recognizeText(imageFile);
    }
}

在這段代碼中，recognizeText(File imageFile)方法負責執(zhí)行對本地文件的OCR任務，而recognizeTextFromUrl(String imageUrl)方法則先將遠程圖片下載到本地，然后再執(zhí)行OCR。

第4部分：建立REST控制器

@RestController
@RequestMapping("/api/ocr")
public class OcrController {

    private final OcrService ocrService;

    // 使用構造器注入OcrService
    public OcrController(OcrService ocrService) {
        this.ocrService = ocrService;
    }

    @PostMapping("/upload")
    public ResponseEntity<String> uploadImage(@RequestParam("file") MultipartFile file) {
        try {
            File convFile = new File(System.getProperty("java.io.tmpdir")+"/"+file.getOriginalFilename());
            file.transferTo(convFile);
            String result = ocrService.recognizeText(convFile);
            return ResponseEntity.ok(result);
        } catch (Exception e) {
            e.printStackTrace();
            return ResponseEntity.badRequest().body("識別發(fā)生錯誤：" + e.getMessage());
        }
    }

    @GetMapping("/recognize-url")
    public ResponseEntity<String> recognizeFromUrl(@RequestParam("imageUrl") String imageUrl) {
        try {
            String result = ocrService.recognizeTextFromUrl(imageUrl);
            return ResponseEntity.ok(result);
        } catch (Exception e) {
            e.printStackTrace();
            return ResponseEntity.badRequest().body("從URL識別發(fā)生錯誤：" + e.getMessage());
        }
    }
}

在這個控制器中，我們創(chuàng)建了兩個端點：/api/ocr/upload用于處理用戶上傳的本地圖片，而/api/ocr/recognize-url則處理給定URL的遠程圖片。