mii/usls
1
0
Fork 0
mirror of https://github.com/mii443/usls.git synced 2025-08-22 23:55:38 +00:00
Code Issues Packages Projects Releases Wiki Activity

0.1.0-beta.1 (#82)

Browse Source
This commit is contained in:
Jamjamjon
2025-04-27 13:01:51 +08:00
committed by GitHub
parent aa25854f7f
commit 80468d9c83
174 changed files with 8413 additions and 6188 deletions
Download Patch File Download Diff File Expand all files Collapse all files

28
examples/depth-pro/main.rs
View File

@ -1,8 +1,9 @@
use anyhow::Result;
use usls::{models::DepthPro, Annotator, DataLoader, Options};
use usls::DataLoader;
use usls::{models::DepthPro, Annotator, Options, Style};
#[derive(argh::FromArgs)]
/// BLIP Example
/// Example
struct Args {
/// device
#[argh(option, default = "String::from(\"cpu:0\")")]
@ -11,10 +12,6 @@ struct Args {
/// dtype
#[argh(option, default = "String::from(\"q4f16\")")]
dtype: String,
/// source image
#[argh(option, default = "String::from(\"images/street.jpg\")")]
source: String,
}
fn main() -> Result<()> {
@ -33,16 +30,23 @@ fn main() -> Result<()> {
let mut model = DepthPro::new(options)?;
// load
let x = [DataLoader::try_read(&args.source)?];
let xs = DataLoader::try_read_n(&["images/street.jpg"])?;
// run
let y = model.forward(&x)?;
let ys = model.forward(&xs)?;
// annotate
let annotator = Annotator::default()
.with_colormap("Turbo")
.with_saveout(model.spec());
annotator.annotate(&x, &y);
let annotator =
Annotator::default().with_mask_style(Style::mask().with_colormap256("turbo".into()));
for (x, y) in xs.iter().zip(ys.iter()) {
annotator.annotate(x, y)?.save(format!(
"{}.jpg",
usls::Dir::Current
.base_dir_with_subs(&["runs", model.spec()])?
.join(usls::timestamp(None))
.display(),
))?;
}
Ok(())
}