inprogress: paddle ocr

yas-genshin/Cargo.toml

@@ -36,3 +36,7 @@ winres = "0.1"
 [[bin]]
 name = "yas_artifact"
 path = "src/bin/artifact_scanner.rs"
+
+[[bin]]
+name = "playground"
+path = "src/bin/playground.rs"

yas-genshin/src/bin/playground.rs

@@ -0,0 +1,14 @@
+use anyhow::Result;
+use image::RgbImage;
+use yas::ocr::{ImageToText, PPOCRChV4RecInfer, ppocr_model};
+use image::io::Reader as ImageReader;
+
+fn main() -> Result<()> {
+    let model: Box<dyn ImageToText<RgbImage>> = Box::new(PPOCRChV4RecInfer::new()?);
+    let image = ImageReader::open(r"E:\rust\yas\item_count.png")?.decode()?;
+    let rgb_image = image.to_rgb8();
+    let result = model.image_to_text(&rgb_image, false)?;
+    println!("{}", result);
+
+    Ok(())
+}

yas/Cargo.toml

@@ -15,7 +15,7 @@ clap = { version = "4.4", features = ["derive"] }
 image = "0.24"
 enigo = "0.1"
 # ref: https://github.com/sonos/tract#tract-opl-version-compatibility
-tract-onnx = "0.17"
+tract-onnx = "0.21.1"
 serde_json = "1.0"
 serde = { version = "1.0", features = ["derive"] }
 regex = "1.5"

yas/src/ocr/mod.rs

@@ -5,3 +5,6 @@ mod paddle_paddle_model;
 pub use yas_model::yas_ocr_model::YasOCRModel;
 pub use yas_model::yas_ocr_model::yas_ocr_model;
 pub use traits::ImageToText;
+pub use paddle_paddle_model::PPOCRModel;
+pub use paddle_paddle_model::PPOCRChV4RecInfer;
+pub use paddle_paddle_model::ppocr_model;