diff --git a/ocr/src/main/java/cn/smartjavaai/ocr/model/plate/translator/CRNNPlateRecTranslator.java b/ocr/src/main/java/cn/smartjavaai/ocr/model/plate/translator/CRNNPlateRecTranslator.java index 1614bc7..7d3a826 100644 --- a/ocr/src/main/java/cn/smartjavaai/ocr/model/plate/translator/CRNNPlateRecTranslator.java +++ b/ocr/src/main/java/cn/smartjavaai/ocr/model/plate/translator/CRNNPlateRecTranslator.java @@ -31,7 +31,8 @@ public NDList processInput(TranslatorContext ctx, Image input) { // Resize to (168, 48) NDArray array = input.toNDArray(manager, Image.Flag.COLOR); array = NDImageUtils.resize(array, 168, 48); - + // RGB → BGR,与 cv2 对齐 + array = array.flip(2); // Normalize array = array.toType(DataType.FLOAT32, false) .div(255f)