mii/usls
1
0
Fork 0
mirror of https://github.com/mii443/usls.git synced 2025-12-03 11:08:20 +00:00
Code Issues Packages Projects Releases Wiki Activity

Options -> ModelConfig

Browse Source
This commit is contained in:
jamjamjon
2025-05-16 15:45:27 +08:00
parent 54045543fc
commit 70e6b2f03e
118 changed files with 1735 additions and 2503 deletions
Download Patch File Download Diff File Expand all files Collapse all files

15
examples/yolo-sam2/main.rs
View File

@@ -1,7 +1,7 @@
use anyhow::Result;
use usls::{
models::{SamPrompt, SAM2, YOLO},
Annotator, DataLoader, Options, Scale, Style,
Annotator, DataLoader, ModelConfig, Scale, Style,
};
#[derive(argh::FromArgs)]
@@ -21,17 +21,14 @@ fn main() -> Result<()> {
let args: Args = argh::from_env();
// build SAM
let (options_encoder, options_decoder) = (
Options::sam2_1_tiny_encoder().commit()?,
Options::sam2_1_tiny_decoder().commit()?,
);
let mut sam = SAM2::new(options_encoder, options_decoder)?;
let mut sam = SAM2::new(ModelConfig::sam2_1_tiny().commit()?)?;
// build YOLOv8
let options_yolo = Options::yolo_detect()
.with_model_scale(Scale::N)
.with_model_version(8.into())
let options_yolo = ModelConfig::yolo_detect()
.with_scale(Scale::N)
.with_version(8.into())
.with_model_device(args.device.as_str().try_into()?)
.auto_yolo_model_file()
.commit()?;
let mut yolo = YOLO::new(options_yolo)?;