ocr图像处理功能开发

zouyu

2025-09-26 3fbbfcc8f509c352c58dc8a126220b49b72ed5a0

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
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
package com.xindao.ocr.smartjavaai.model.common.direction;
 
import ai.djl.inference.Predictor;
import ai.djl.modality.cv.Image;
import com.xindao.ocr.smartjavaai.config.DirectionModelConfig;
import com.xindao.ocr.smartjavaai.entity.DirectionInfo;
import com.xindao.ocr.smartjavaai.entity.OcrBox;
import com.xindao.ocr.smartjavaai.entity.OcrItem;
import com.xindao.ocr.smartjavaai.model.common.detect.OcrCommonDetModel;
import org.apache.commons.pool2.impl.GenericObjectPool;
import org.opencv.core.Mat;
 
import java.awt.image.BufferedImage;
import java.util.List;
 
/**
 * OCR 文本方向分类模型
 * @author dwj
 */
public interface OcrDirectionModel extends AutoCloseable{
 
    default void setTextDetModel(OcrCommonDetModel detModel){
        throw new UnsupportedOperationException("默认不支持该功能");
    }
 
    default OcrCommonDetModel getTextDetModel(){
        throw new UnsupportedOperationException("默认不支持该功能");
    }
 
    /**
     * 加载模型
     * @param config
     */
    void loadModel(DirectionModelConfig config); // 加载模型
 
    /**
     * 文本方向检测
     * @param imagePath 图片路径
     * @return
     */
    default List<OcrItem> detect(String imagePath) {
        throw new UnsupportedOperationException("默认不支持该功能");
    }
 
 
    /**
     * 文本方向检测
     * @param image BufferedImage
     * @return
     */
    default List<OcrItem> detect(BufferedImage image) {
        throw new UnsupportedOperationException("默认不支持该功能");
    }
 
 
    /**
     * 文本方向检测
     * @param imageData 图片字节数组
     * @return
     */
    default List<OcrItem> detect(byte[] imageData) {
        throw new UnsupportedOperationException("默认不支持该功能");
    }
 
    /**
     * 文本方向检测
     * @param image
     * @return
     */
    default List<OcrItem> detect(Image image) {
        throw new UnsupportedOperationException("默认不支持该功能");
    }
 
 
    /**
     * 文本方向检测（基于检测结果）
     * @param boxList
     * @param srcMat
     * @param manager
     * @return
     */
    default List<OcrItem> detect(List<OcrBox> boxList, Mat srcMat) {
        throw new UnsupportedOperationException("默认不支持该功能");
    }
 
    default List<List<OcrItem>> batchDetect(List<List<OcrBox>> boxList, List<Mat> srcMatList) {
        throw new UnsupportedOperationException("默认不支持该功能");
    }
 
    /**
     * 检测并绘制结果
     * @param imagePath 图片输入路径（包含文件名称）
     * @param outputPath 图片输出路径（包含文件名称）
     */
    default void detectAndDraw(String imagePath, String outputPath) {
        throw new UnsupportedOperationException("默认不支持该功能");
    }
 
    /**
     * 检测并绘制结果
     * @param sourceImage
     * @return
     */
    default BufferedImage detectAndDraw(BufferedImage sourceImage){
        throw new UnsupportedOperationException("默认不支持该功能");
    }
 
    default GenericObjectPool<Predictor<Image, DirectionInfo>> getPool() {
        throw new UnsupportedOperationException("默认不支持该功能");
    }
 
}