mii/usls
1
0
Fork 0
mirror of https://github.com/mii443/usls.git synced 2025-08-22 15:45:41 +00:00
Code Issues Packages Projects Releases Wiki Activity

Add PPOCRv5 DET and REC models (#98)

Browse Source 
* Add PPOCRv5 DET and REC models

* Add Text Struct
This commit is contained in:
Jamjamjon
2025-05-21 13:29:17 +08:00
committed by GitHub
parent a2dde94a3a
commit cf21eabcaa
16 changed files with 137 additions and 38 deletions
Download Patch File Download Diff File Expand all files Collapse all files

20
examples/db/main.rs
View File

@ -28,9 +28,13 @@ struct Args {
#[argh(option, default = "false")]
show_hbbs_conf: bool,
/// show mbrs confidence
/// show obbs confidence
#[argh(option, default = "false")]
show_obbs_conf: bool,
/// show polygons confidence
#[argh(option, default = "false")]
show_polygons_conf: bool,
}
fn main() -> Result<()> {
@ -43,7 +47,7 @@ fn main() -> Result<()> {
// build model
let config = match &args.model {
Some(m) => Config::db().with_model_file(m),
None => Config::ppocr_det_v4_ch().with_model_dtype(args.dtype.as_str().try_into()?),
None => Config::ppocr_det_v5_mobile().with_model_dtype(args.dtype.as_str().try_into()?),
}
.with_device_all(args.device.as_str().try_into()?)
.commit()?;
@ -66,16 +70,16 @@ fn main() -> Result<()> {
.with_polygon_style(
Style::polygon()
.with_visible(true)
.with_text_visible(false)
.show_confidence(true)
.show_id(true)
.show_name(true)
.with_text_visible(true)
.show_confidence(args.show_polygons_conf)
.show_id(false)
.show_name(false)
.with_color(usls::StyleColors::default().with_outline([255, 105, 180, 255].into())),
)
.with_hbb_style(
Style::hbb()
.with_visible(args.show_hbbs)
.with_text_visible(false)
.with_text_visible(true)
.with_thickness(1)
.show_confidence(args.show_hbbs_conf)
.show_id(false)
@ -84,7 +88,7 @@ fn main() -> Result<()> {
.with_obb_style(
Style::obb()
.with_visible(args.show_obbs)
.with_text_visible(false)
.with_text_visible(true)
.show_confidence(args.show_obbs_conf)
.show_id(false)
.show_name(false),