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

Fix errors related to GroundingDino class_names (#71)

Browse Source
This commit is contained in:
Jamjamjon
2025-03-29 15:34:18 +08:00
committed by GitHub
parent 293c7d2e38
commit 118690402d
3 changed files with 113 additions and 87 deletions
Download Patch File Download Diff File Expand all files Collapse all files

26
examples/grounding-dino/main.rs
View File

@ -20,18 +20,18 @@ struct Args {
#[argh(
option,
default = "vec![
String::from(\"person\"),
String::from(\"hand\"),
String::from(\"shoes\"),
String::from(\"bus\"),
String::from(\"dog\"),
String::from(\"cat\"),
String::from(\"sign\"),
String::from(\"tie\"),
String::from(\"monitor\"),
String::from(\"glasses\"),
String::from(\"tree\"),
String::from(\"head\"),
String::from(\"person\"),
String::from(\"a hand\"),
String::from(\"a shoe\"),
String::from(\"bus\"),
String::from(\"dog\"),
String::from(\"cat\"),
String::from(\"sign\"),
String::from(\"tie\"),
String::from(\"monitor\"),
String::from(\"glasses\"),
String::from(\"tree\"),
String::from(\"head\"),
]"
)]
labels: Vec<String>,
@ -49,6 +49,8 @@ fn main() -> Result<()> {
.with_model_dtype(args.dtype.as_str().try_into()?)
.with_model_device(args.device.as_str().try_into()?)
.with_text_names(&args.labels.iter().map(|x| x.as_str()).collect::<Vec<_>>())
.with_class_confs(&[0.25])
.with_text_confs(&[0.25])
.commit()?;
let mut model = GroundingDINO::new(options)?;