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 YOLOPv2 & Face-Parsing model (#3)

Browse Source 
* Add YOLOP and face parsing model
This commit is contained in:
Jamjamjon
2024-04-14 15:15:59 +08:00
committed by GitHub
parent ead175234c
commit 51b75e9a21
63 changed files with 1168 additions and 885 deletions
Download Patch File Download Diff File Expand all files Collapse all files

21
examples/db/README.md
View File

@ -4,25 +4,10 @@
cargo run -r --example db
```
## Or you can manully
## ONNX Model
### 1. Donwload ONNX Model
[ppocr-v3-db-dyn](https://github.com/jamjamjon/assets/releases/download/v0.0.1/ppocr-v3-db-dyn.onnx)
[ppocr-v4-db-dyn](https://github.com/jamjamjon/assets/releases/download/v0.0.1/ppocr-v4-db-dyn.onnx)
### 2. Specify the ONNX model path in `main.rs`
```Rust
let options = Options::default()
.with_model("ONNX_PATH") // <= modify this
```
### 3. Run
```bash
cargo run -r --example db
```
- [ppocr-v3-db-dyn](https://github.com/jamjamjon/assets/releases/download/v0.0.1/ppocr-v3-db-dyn.onnx)
- [ppocr-v4-db-dyn](https://github.com/jamjamjon/assets/releases/download/v0.0.1/ppocr-v4-db-dyn.onnx)
### Speed test

2
examples/db/main.rs
View File

@ -22,9 +22,9 @@ fn main() -> Result<(), Box<dyn std::error::Error>> {
// annotate
let annotator = Annotator::default()
.with_polygon_color([255u8, 0u8, 0u8])
.without_name(true)
.without_polygons(false)
.with_mask_alpha(0)
.without_bboxes(false)
.with_saveout("DB-Text-Detection");
annotator.annotate(&x, &y);