mirror of
https://github.com/mii443/usls.git
synced 2025-08-22 15:45:41 +00:00
0adddd3bbd
- Added `Hub` for resource management - Updated `DataLoader` to support video and streaming - Updated `CI` - Replaced `println!` with `tracing` for logging
42 lines
1.5 KiB
Rust
42 lines
1.5 KiB
Rust
use usls::{models::GroundingDINO, Annotator, DataLoader, Options};
|
|
|
|
fn main() -> Result<(), Box<dyn std::error::Error>> {
|
|
let opts = Options::default()
|
|
.with_i00((1, 1, 4).into())
|
|
.with_i02((640, 800, 1200).into())
|
|
.with_i03((640, 1200, 1200).into())
|
|
.with_i10((1, 1, 4).into())
|
|
.with_i11((256, 256, 512).into())
|
|
.with_i20((1, 1, 4).into())
|
|
.with_i21((256, 256, 512).into())
|
|
.with_i30((1, 1, 4).into())
|
|
.with_i31((256, 256, 512).into())
|
|
.with_i40((1, 1, 4).into())
|
|
.with_i41((256, 256, 512).into())
|
|
.with_i50((1, 1, 4).into())
|
|
.with_i51((256, 256, 512).into())
|
|
.with_i52((256, 256, 512).into())
|
|
.with_model("grounding-dino/swint-ogc-dyn-u8.onnx")? // TODO: current onnx model does not support bs > 1
|
|
// .with_model("grounding-dino/swint-ogc-dyn-f32.onnx")?
|
|
.with_tokenizer("grounding-dino/tokenizer.json")?
|
|
.with_confs(&[0.2])
|
|
.with_profile(false);
|
|
let mut model = GroundingDINO::new(opts)?;
|
|
|
|
// Load images and set class names
|
|
let x = [DataLoader::try_read("images/bus.jpg")?];
|
|
let texts = [
|
|
"person", "hand", "shoes", "bus", "dog", "cat", "sign", "tie", "monitor", "window",
|
|
"glasses", "tree", "head",
|
|
];
|
|
|
|
// Run and annotate
|
|
let y = model.run(&x, &texts)?;
|
|
let annotator = Annotator::default()
|
|
.with_bboxes_thickness(4)
|
|
.with_saveout("GroundingDINO");
|
|
annotator.annotate(&x, &y);
|
|
|
|
Ok(())
|
|
}
|