//这个写法其实不太好，绝对地址靠谱点
static String DET_PATH = "src/main/java/org/example/models/ch_ptocr_det_infer.pt";
static String REC_PATH = "src/main/java/org/example/models/ch_ptocr_rec_infer.pt";
//下面代码会有些异常需要处理，我就不写进去了
/**
 * DET模型构建
 */
Criteria<Image, DetectedObjects> criteria_det = Criteria.builder()
        .setTypes(Image.class, DetectedObjects.class)
        .optModelPath(Paths.get(DET_PATH))
    	//加载Translator
        .optTranslator(new PpWordDetectionTranslator(new ConcurrentHashMap<String, String>()))
        .build();
ZooModel<Image, DetectedObjects> detectionModel=criteria_det.loadModel();

/**
 * REC模型构建
 */
 Criteria<Image, String> criteria_rec = Criteria.builder()
        .setTypes(Image.class, String.class)
        .optModelPath(Paths.get(REC_PATH))
        .optTranslator(new PpWordRecognitionTranslator())
        .optProgress(new ProgressBar())
     	.build();
ZooModel<Image, String> recognitionModel=criteria_rec.loadModel();
/**
 * 两个Predictor生成
 */
Predictor<Image, DetectedObjects> detector = detectionModel.newPredictor();
Predictor<Image, String> recognizer = recognitionModel.newPredictor();

/**
 * 加载图片，直接传递路径即可
 */
Image img = ImageFactory.getInstance().fromFile(Paths.get(path));

/**
 * 文字区域检测
 */
DetectedObjects detectedObj = detector.predict(img);
Image newImage = img.duplicate();
newImage.drawBoundingBoxes(detectedObj);
newImage.getWrappedImage();

/**
 * 获取分割出来的文字区域列表,并识别返回文本
 */
List<DetectedObjects.DetectedObject> boxes = detectedObj.items();
StringBuilder sb = new StringBuilder();
System.out.println(boxes.size());
for (int i = 0; i < boxes.size(); i++) {
    Image subImage = getSubImage(img, boxes.get(i).getBoundingBox());
    subImage.getWrappedImage();
    String predict = recognizer.predict(subImage);
    sb.append(predict);
}
String result = sb.toString();

public static Image getSubImage(Image img, BoundingBox box) {
    Rectangle rect = box.getBounds();
    double[] extended = extendRect(rect.getX(), rect.getY(), rect.getWidth(), rect.getHeight());
    int width = img.getWidth();
    int height = img.getHeight();
    int[] recovered = {
            (int) (extended[0] * width),
            (int) (extended[1] * height),
            (int) (extended[2] * width),
            (int) (extended[3] * height)
    };
    return img.getSubImage(recovered[0], recovered[1], recovered[2], recovered[3]);
}

public static double[] extendRect(double xmin, double ymin, double width, double height) {
    double centerx = xmin + width / 2;
    double centery = ymin + height / 2;
    if (width > height) {
        width += height * 1.6;
        height *= 2.6;
    } else {
        height += width * 1.6;
        width *= 2.6;
    }
    double newX = centerx - width / 2 < 0 ? 0 : centerx - width / 2;
    double newY = centery - height / 2 < 0 ? 0 : centery - height / 2;
    double newWidth = newX + width > 1 ? 1 - newX : width;
    double newHeight = newY + height > 1 ? 1 - newY : height;
    return new double[] {newX, newY, newWidth, newHeight};
}
public static Image rotateImg(Image image) {
    try (NDManager manager = NDManager.newBaseManager()) {
        NDArray rotated = NDImageUtils.rotate90(image.toNDArray(manager), 1);
        return ImageFactory.getInstance().fromNDArray(rotated);
    }
}