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

Adding external data file support for ONNX models (#104)

Browse Source
This commit is contained in:
Jamjamjon
2025-05-28 17:23:33 +08:00
committed by GitHub
parent cfb92105ee
commit 370437b67a
2 changed files with 14 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

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

@ -5,7 +5,7 @@ use usls::{models::GroundingDINO, Annotator, Config, DataLoader};
/// Example /// Example
struct Args { struct Args {
/// dtype /// dtype
#[argh(option, default = "String::from(\"fp16\")")] #[argh(option, default = "String::from(\"q8\")")]
dtype: String, dtype: String,
/// device /// device

13
src/utils/ort_config.rs
View File

@ -72,6 +72,19 @@ impl ORTConfig {
let stem = try_fetch_file_stem(&self.file)?; let stem = try_fetch_file_stem(&self.file)?;
self.spec = format!("{}/{}", name, stem); self.spec = format!("{}/{}", name, stem);
self.file = Hub::default().try_fetch(&format!("{}/{}", name, self.file))?; self.file = Hub::default().try_fetch(&format!("{}/{}", name, self.file))?;
// try fetch external data file if it exists
match Hub::default().try_fetch(&format!("{}_data", self.file)) {
Ok(external_data_file) => {
log::debug!(
"Successfully fetched external data file: {}",
external_data_file
);
}
Err(_) => {
log::debug!("No external data file found for model {}", self.file);
}
}
} }
} }
} }