mirror of
https://github.com/mii443/usls.git
synced 2025-08-22 15:45:41 +00:00
6ace97f09f
* Add `Viewer` for image-show * Add .`with_ixx()` to simplify model building with `Options` * Update `MinOptMax`
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_ixx(0, 0, (1, 1, 4).into())
|
|
.with_ixx(0, 2, (640, 800, 1200).into())
|
|
.with_ixx(0, 3, (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(())
|
|
}
|