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

update

Browse Source
This commit is contained in:
jamjamjon
2025-05-20 17:01:27 +08:00
parent 26de63d239
commit d3c738b5cf
110 changed files with 298 additions and 264 deletions
Download Patch File Download Diff File Expand all files Collapse all files

4
examples/ben2/main.rs
View File

@@ -1,4 +1,4 @@
use usls::{models::RMBG, Annotator, DataLoader, ModelConfig};
use usls::{models::RMBG, Annotator, Config, DataLoader};
#[derive(argh::FromArgs)]
/// Example
@@ -20,7 +20,7 @@ fn main() -> anyhow::Result<()> {
let args: Args = argh::from_env();
// build model
let config = ModelConfig::ben2_base()
let config = Config::ben2_base()
.with_model_dtype(args.dtype.as_str().try_into()?)
.with_model_device(args.device.as_str().try_into()?)
.commit()?;

4
examples/blip/main.rs
View File

@@ -1,4 +1,4 @@
use usls::{models::Blip, DataLoader, ModelConfig};
use usls::{models::Blip, Config, DataLoader};
#[derive(argh::FromArgs)]
/// BLIP Example
@@ -20,7 +20,7 @@ fn main() -> anyhow::Result<()> {
let args: Args = argh::from_env();
// build model
let config = ModelConfig::blip_v1_base_caption()
let config = Config::blip_v1_base_caption()
.with_device_all(args.device.as_str().try_into()?)
.commit()?;
let mut model = Blip::new(config)?;

12
examples/classifier/main.rs
View File

@@ -1,4 +1,4 @@
use usls::{models::ImageClassifier, Annotator, DataLoader, ModelConfig};
use usls::{models::ImageClassifier, Annotator, Config, DataLoader};
#[derive(argh::FromArgs)]
/// Example
@@ -37,11 +37,11 @@ fn main() -> anyhow::Result<()> {
// build model
let config = match args.model.to_lowercase().as_str() {
"beit" => ModelConfig::beit_base(),
"convnext" => ModelConfig::convnext_v2_atto(),
"deit" => ModelConfig::deit_tiny_distill(),
"fastvit" => ModelConfig::fastvit_t8_distill(),
"mobileone" => ModelConfig::mobileone_s0(),
"beit" => Config::beit_base(),
"convnext" => Config::convnext_v2_atto(),
"deit" => Config::deit_tiny_distill(),
"fastvit" => Config::fastvit_t8_distill(),
"mobileone" => Config::mobileone_s0(),
_ => anyhow::bail!("Unsupported model: {}", args.model),
};

4
examples/clip/main.rs
View File

@@ -1,5 +1,5 @@
use anyhow::Result;
use usls::{models::Clip, DataLoader, ModelConfig, Ops};
use usls::{models::Clip, Config, DataLoader, Ops};
#[derive(argh::FromArgs)]
/// CLIP Example
@@ -17,7 +17,7 @@ fn main() -> Result<()> {
let args: Args = argh::from_env();
// build model
let config = ModelConfig::jina_clip_v1()
let config = Config::jina_clip_v1()
.with_device_all(args.device.as_str().try_into()?)
.commit()?;
let mut model = Clip::new(config)?;

4
examples/d-fine/main.rs
View File

@@ -1,5 +1,5 @@
use anyhow::Result;
use usls::{models::RTDETR, Annotator, DataLoader, ModelConfig};
use usls::{models::RTDETR, Annotator, Config, DataLoader};
fn main() -> Result<()> {
tracing_subscriber::fmt()
@@ -8,7 +8,7 @@ fn main() -> Result<()> {
.init();
// config
let mut model = RTDETR::new(ModelConfig::d_fine_n_coco().commit()?)?;
let mut model = RTDETR::new(Config::d_fine_n_coco().commit()?)?;
// load
let xs = DataLoader::try_read_n(&["./assets/bus.jpg"])?;

6
examples/db/main.rs
View File

@@ -1,5 +1,5 @@
use anyhow::Result;
use usls::{models::DB, Annotator, DataLoader, ModelConfig, Style};
use usls::{models::DB, Annotator, Config, DataLoader, Style};
#[derive(argh::FromArgs)]
/// Example
@@ -42,8 +42,8 @@ fn main() -> Result<()> {
// build model
let config = match &args.model {
Some(m) => ModelConfig::db().with_model_file(m),
None => ModelConfig::ppocr_det_v4_ch().with_model_dtype(args.dtype.as_str().try_into()?),
Some(m) => Config::db().with_model_file(m),
None => Config::ppocr_det_v4_ch().with_model_dtype(args.dtype.as_str().try_into()?),
}
.with_device_all(args.device.as_str().try_into()?)
.commit()?;

4
examples/deim/main.rs
View File

@@ -1,5 +1,5 @@
use anyhow::Result;
use usls::{models::RTDETR, Annotator, DataLoader, ModelConfig};
use usls::{models::RTDETR, Annotator, Config, DataLoader};
fn main() -> Result<()> {
tracing_subscriber::fmt()
@@ -8,7 +8,7 @@ fn main() -> Result<()> {
.init();
// config
let mut model = RTDETR::new(ModelConfig::deim_dfine_s_coco().commit()?)?;
let mut model = RTDETR::new(Config::deim_dfine_s_coco().commit()?)?;
// load
let xs = DataLoader::try_read_n(&["./assets/bus.jpg"])?;

4
examples/depth-anything/main.rs
View File

@@ -1,5 +1,5 @@
use anyhow::Result;
use usls::{models::DepthAnything, Annotator, DataLoader, ModelConfig, Style};
use usls::{models::DepthAnything, Annotator, Config, DataLoader, Style};
fn main() -> Result<()> {
tracing_subscriber::fmt()
@@ -8,7 +8,7 @@ fn main() -> Result<()> {
.init();
// build model
let mut model = DepthAnything::new(ModelConfig::depth_anything_v2_small().commit()?)?;
let mut model = DepthAnything::new(Config::depth_anything_v2_small().commit()?)?;
// load
let xs = DataLoader::try_read_n(&["images/street.jpg"])?;

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

@@ -1,6 +1,6 @@
use anyhow::Result;
use usls::DataLoader;
use usls::{models::DepthPro, Annotator, ModelConfig, Style};
use usls::{models::DepthPro, Annotator, Config, Style};
#[derive(argh::FromArgs)]
/// Example
@@ -23,7 +23,7 @@ fn main() -> Result<()> {
let args: Args = argh::from_env();
// model
let config = ModelConfig::depth_pro()
let config = Config::depth_pro()
.with_model_dtype(args.dtype.as_str().try_into()?)
.with_model_device(args.device.as_str().try_into()?)
.commit()?;

4
examples/dinov2/main.rs
View File

@@ -1,5 +1,5 @@
use anyhow::Result;
use usls::{models::DINOv2, DataLoader, ModelConfig};
use usls::{models::DINOv2, Config, DataLoader};
fn main() -> Result<()> {
tracing_subscriber::fmt()
@@ -11,7 +11,7 @@ fn main() -> Result<()> {
let xs = DataLoader::try_read_n(&["./assets/bus.jpg", "./assets/bus.jpg"])?;
// model
let config = ModelConfig::dinov2_small()
let config = Config::dinov2_small()
.with_batch_size_all(xs.len())
.commit()?;
let mut model = DINOv2::new(config)?;

4
examples/doclayout-yolo/main.rs
View File

@@ -1,5 +1,5 @@
use anyhow::Result;
use usls::{models::YOLO, Annotator, DataLoader, ModelConfig};
use usls::{models::YOLO, Annotator, Config, DataLoader};
#[derive(argh::FromArgs)]
/// Example
@@ -18,7 +18,7 @@ fn main() -> Result<()> {
let args: Args = argh::from_env();
// build model
let config = ModelConfig::doclayout_yolo_docstructbench()
let config = Config::doclayout_yolo_docstructbench()
.with_model_device(args.device.as_str().try_into()?)
.commit()?;
let mut model = YOLO::new(config)?;

8
examples/fast/main.rs
View File

@@ -1,5 +1,5 @@
use anyhow::Result;
use usls::{models::DB, Annotator, DataLoader, ModelConfig, Scale, Style};
use usls::{models::DB, Annotator, Config, DataLoader, Scale, Style};
#[derive(argh::FromArgs)]
/// Example
@@ -27,9 +27,9 @@ fn main() -> Result<()> {
// build model
let config = match args.scale.as_str().try_into()? {
Scale::T => ModelConfig::fast_tiny(),
Scale::S => ModelConfig::fast_small(),
Scale::B => ModelConfig::fast_base(),
Scale::T => Config::fast_tiny(),
Scale::S => Config::fast_small(),
Scale::B => Config::fast_base(),
_ => unimplemented!("Unsupported model scale: {:?}. Try b, s, t.", args.scale),
};
let mut model = DB::new(

4
examples/fastsam/main.rs
View File

@@ -1,5 +1,5 @@
use anyhow::Result;
use usls::{models::YOLO, Annotator, DataLoader, ModelConfig};
use usls::{models::YOLO, Annotator, Config, DataLoader};
#[derive(argh::FromArgs)]
/// Example
@@ -22,7 +22,7 @@ fn main() -> Result<()> {
let args: Args = argh::from_env();
// build model
let config = ModelConfig::fastsam_s()
let config = Config::fastsam_s()
.with_model_dtype(args.dtype.as_str().try_into()?)
.with_model_device(args.device.as_str().try_into()?)
.commit()?;

4
examples/florence2/main.rs
View File

@@ -1,5 +1,5 @@
use anyhow::Result;
use usls::{models::Florence2, Annotator, DataLoader, ModelConfig, Style, Task};
use usls::{models::Florence2, Annotator, Config, DataLoader, Style, Task};
#[derive(argh::FromArgs)]
/// Example
@@ -25,7 +25,7 @@ fn main() -> Result<()> {
let xs = DataLoader::try_read_n(&["images/green-car.jpg", "assets/bus.jpg"])?;
// build model
let config = ModelConfig::florence2_base()
let config = Config::florence2_base()
.with_dtype_all(args.dtype.as_str().try_into()?)
.with_device_all(args.device.as_str().try_into()?)
.with_batch_size_all(xs.len())

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

@@ -1,5 +1,5 @@
use anyhow::Result;
use usls::{models::GroundingDINO, Annotator, DataLoader, ModelConfig};
use usls::{models::GroundingDINO, Annotator, Config, DataLoader};
#[derive(argh::FromArgs)]
/// Example
@@ -45,7 +45,7 @@ fn main() -> Result<()> {
let args: Args = argh::from_env();
let config = ModelConfig::grounding_dino_tiny()
let config = Config::grounding_dino_tiny()
.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<_>>())

8
examples/linknet/main.rs
View File

@@ -1,6 +1,6 @@
use anyhow::Result;
use usls::DataLoader;
use usls::{models::DB, Annotator, ModelConfig, Scale, Style};
use usls::{models::DB, Annotator, Config, Scale, Style};
#[derive(argh::FromArgs)]
/// Example
@@ -28,9 +28,9 @@ fn main() -> Result<()> {
// build model
let config = match args.scale.as_str().try_into()? {
Scale::T => ModelConfig::linknet_r18(),
Scale::S => ModelConfig::linknet_r34(),
Scale::B => ModelConfig::linknet_r50(),
Scale::T => Config::linknet_r18(),
Scale::S => Config::linknet_r34(),
Scale::B => Config::linknet_r50(),
_ => unimplemented!("Unsupported model scale: {:?}. Try b, s, t.", args.scale),
};
let mut model = DB::new(

4
examples/modnet/main.rs
View File

@@ -1,4 +1,4 @@
use usls::{models::MODNet, Annotator, DataLoader, ModelConfig};
use usls::{models::MODNet, Annotator, Config, DataLoader};
fn main() -> anyhow::Result<()> {
tracing_subscriber::fmt()
@@ -7,7 +7,7 @@ fn main() -> anyhow::Result<()> {
.init();
// build model
let mut model = MODNet::new(ModelConfig::modnet_photographic().commit()?)?;
let mut model = MODNet::new(Config::modnet_photographic().commit()?)?;
// load image
let xs = DataLoader::try_read_n(&["images/liuyifei.png"])?;

6
examples/moondream2/main.rs
View File

@@ -1,5 +1,5 @@
use anyhow::Result;
use usls::{models::Moondream2, Annotator, DataLoader, ModelConfig, Scale, Task};
use usls::{models::Moondream2, Annotator, Config, DataLoader, Scale, Task};
#[derive(argh::FromArgs)]
/// Example
@@ -40,8 +40,8 @@ fn main() -> Result<()> {
// build model
let config = match args.scale.as_str().try_into()? {
Scale::Billion(0.5) => ModelConfig::moondream2_0_5b(),
Scale::Billion(2.) => ModelConfig::moondream2_2b(),
Scale::Billion(0.5) => Config::moondream2_0_5b(),
Scale::Billion(2.) => Config::moondream2_2b(),
_ => unimplemented!(),
}
.with_dtype_all(args.dtype.as_str().try_into()?)

4
examples/owlv2/main.rs
View File

@@ -1,6 +1,6 @@
use anyhow::Result;
use usls::DataLoader;
use usls::{models::OWLv2, Annotator, ModelConfig};
use usls::{models::OWLv2, Annotator, Config};
#[derive(argh::FromArgs)]
/// Example
@@ -47,7 +47,7 @@ fn main() -> Result<()> {
let args: Args = argh::from_env();
// config
let config = ModelConfig::owlv2_base_ensemble()
let config = Config::owlv2_base_ensemble()
// owlv2_base()
.with_model_dtype(args.dtype.as_str().try_into()?)
.with_model_device(args.device.as_str().try_into()?)

4
examples/picodet-layout/main.rs
View File

@@ -1,6 +1,6 @@
use anyhow::Result;
use usls::DataLoader;
use usls::{models::PicoDet, Annotator, ModelConfig};
use usls::{models::PicoDet, Annotator, Config};
fn main() -> Result<()> {
tracing_subscriber::fmt()
@@ -9,7 +9,7 @@ fn main() -> Result<()> {
.init();
// config
let config = ModelConfig::picodet_layout_1x().commit()?;
let config = Config::picodet_layout_1x().commit()?;
// picodet_l_layout_3cls()
// picodet_l_layout_17cls()
let mut model = PicoDet::new(config)?;

4
examples/rfdetr/main.rs
View File

@@ -1,5 +1,5 @@
use anyhow::Result;
use usls::{models::RFDETR, Annotator, DataLoader, ModelConfig};
use usls::{models::RFDETR, Annotator, Config, DataLoader};
fn main() -> Result<()> {
tracing_subscriber::fmt()
@@ -8,7 +8,7 @@ fn main() -> Result<()> {
.init();
// config
let mut model = RFDETR::new(ModelConfig::rfdetr_base().commit()?)?;
let mut model = RFDETR::new(Config::rfdetr_base().commit()?)?;
// load
let xs = DataLoader::try_read_n(&["./assets/bus.jpg"])?;

6
examples/rmbg/main.rs
View File

@@ -1,4 +1,4 @@
use usls::{models::RMBG, Annotator, DataLoader, ModelConfig};
use usls::{models::RMBG, Annotator, Config, DataLoader};
#[derive(argh::FromArgs)]
/// Example
@@ -24,8 +24,8 @@ fn main() -> anyhow::Result<()> {
let args: Args = argh::from_env();
let config = match args.ver {
1.4 => ModelConfig::rmbg1_4(),
2.0 => ModelConfig::rmbg2_0(),
1.4 => Config::rmbg1_4(),
2.0 => Config::rmbg2_0(),
_ => unreachable!("Unsupported version"),
};

4
examples/rtdetr/main.rs
View File

@@ -1,5 +1,5 @@
use anyhow::Result;
use usls::{models::RTDETR, Annotator, DataLoader, ModelConfig};
use usls::{models::RTDETR, Annotator, Config, DataLoader};
fn main() -> Result<()> {
tracing_subscriber::fmt()
@@ -8,7 +8,7 @@ fn main() -> Result<()> {
.init();
// config
let config = ModelConfig::rtdetr_v2_s_coco().commit()?;
let config = Config::rtdetr_v2_s_coco().commit()?;
// rtdetr_v1_r18vd_coco()
// rtdetr_v2_ms_coco()
// rtdetr_v2_m_coco()

4
examples/rtmo/main.rs
View File

@@ -1,5 +1,5 @@
use anyhow::Result;
use usls::{models::RTMO, Annotator, DataLoader, ModelConfig, Style, SKELETON_COCO_19};
use usls::{models::RTMO, Annotator, Config, DataLoader, Style, SKELETON_COCO_19};
fn main() -> Result<()> {
tracing_subscriber::fmt()
@@ -8,7 +8,7 @@ fn main() -> Result<()> {
.init();
// build model
let mut model = RTMO::new(ModelConfig::rtmo_s().commit()?)?;
let mut model = RTMO::new(Config::rtmo_s().commit()?)?;
// load image
let xs = DataLoader::try_read_n(&["./assets/bus.jpg"])?;

16
examples/sam/main.rs
View File

@@ -1,7 +1,7 @@
use anyhow::Result;
use usls::{
models::{SamKind, SamPrompt, SAM},
Annotator, DataLoader, ModelConfig, Scale,
Annotator, Config, DataLoader, Scale,
};
#[derive(argh::FromArgs)]
@@ -29,16 +29,16 @@ fn main() -> Result<()> {
let args: Args = argh::from_env();
// Build model
let config = match args.kind.as_str().try_into()? {
SamKind::Sam => ModelConfig::sam_v1_base(),
SamKind::Sam => Config::sam_v1_base(),
SamKind::Sam2 => match args.scale.as_str().try_into()? {
Scale::T => ModelConfig::sam2_tiny(),
Scale::S => ModelConfig::sam2_small(),
Scale::B => ModelConfig::sam2_base_plus(),
Scale::T => Config::sam2_tiny(),
Scale::S => Config::sam2_small(),
Scale::B => Config::sam2_base_plus(),
_ => unimplemented!("Unsupported model scale: {:?}. Try b, s, t.", args.scale),
},
SamKind::MobileSam => ModelConfig::mobile_sam_tiny(),
SamKind::SamHq => ModelConfig::sam_hq_tiny(),
SamKind::EdgeSam => ModelConfig::edge_sam_3x(),
SamKind::MobileSam => Config::mobile_sam_tiny(),
SamKind::SamHq => Config::sam_hq_tiny(),
SamKind::EdgeSam => Config::edge_sam_3x(),
}
.with_device_all(args.device.as_str().try_into()?)
.commit()?;

10
examples/sam2/main.rs
View File

@@ -1,7 +1,7 @@
use anyhow::Result;
use usls::{
models::{SamPrompt, SAM2},
Annotator, DataLoader, ModelConfig, Scale,
Annotator, Config, DataLoader, Scale,
};
#[derive(argh::FromArgs)]
@@ -26,10 +26,10 @@ fn main() -> Result<()> {
// Build model
let config = match args.scale.as_str().try_into()? {
Scale::T => ModelConfig::sam2_1_tiny(),
Scale::S => ModelConfig::sam2_1_small(),
Scale::B => ModelConfig::sam2_1_base_plus(),
Scale::L => ModelConfig::sam2_1_large(),
Scale::T => Config::sam2_1_tiny(),
Scale::S => Config::sam2_1_small(),
Scale::B => Config::sam2_1_base_plus(),
Scale::L => Config::sam2_1_large(),
_ => unimplemented!("Unsupported model scale: {:?}. Try b, s, t, l.", args.scale),
}
.with_device_all(args.device.as_str().try_into()?)

4
examples/sapiens/main.rs
View File

@@ -1,5 +1,5 @@
use anyhow::Result;
use usls::{models::Sapiens, Annotator, DataLoader, ModelConfig};
use usls::{models::Sapiens, Annotator, Config, DataLoader};
#[derive(argh::FromArgs)]
/// Example
@@ -17,7 +17,7 @@ fn main() -> Result<()> {
let args: Args = argh::from_env();
// build
let config = ModelConfig::sapiens_seg_0_3b()
let config = Config::sapiens_seg_0_3b()
.with_model_device(args.device.as_str().try_into()?)
.commit()?;
let mut model = Sapiens::new(config)?;

4
examples/slanet/main.rs
View File

@@ -1,5 +1,5 @@
use anyhow::Result;
use usls::{models::SLANet, Annotator, Color, DataLoader, ModelConfig};
use usls::{models::SLANet, Annotator, Color, Config, DataLoader};
#[derive(argh::FromArgs)]
/// Example
@@ -26,7 +26,7 @@ fn main() -> Result<()> {
let args: Args = argh::from_env();
// build model
let config = ModelConfig::slanet_lcnet_v2_mobile_ch()
let config = Config::slanet_lcnet_v2_mobile_ch()
.with_model_device(args.device.as_str().try_into()?)
.with_model_dtype(args.dtype.as_str().try_into()?)
.commit()?;

6
examples/smolvlm/main.rs
View File

@@ -1,5 +1,5 @@
use anyhow::Result;
use usls::{models::SmolVLM, DataLoader, ModelConfig, Scale};
use usls::{models::SmolVLM, Config, DataLoader, Scale};
#[derive(argh::FromArgs)]
/// Example
@@ -30,8 +30,8 @@ fn main() -> Result<()> {
// build model
let config = match args.scale.as_str().try_into()? {
Scale::Million(256.) => ModelConfig::smolvlm_256m(),
Scale::Million(500.) => ModelConfig::smolvlm_500m(),
Scale::Million(256.) => Config::smolvlm_256m(),
Scale::Million(500.) => Config::smolvlm_500m(),
_ => unimplemented!(),
}
.with_device_all(args.device.as_str().try_into()?)

4
examples/svtr/main.rs
View File

@@ -1,5 +1,5 @@
use anyhow::Result;
use usls::{models::SVTR, DataLoader, ModelConfig};
use usls::{models::SVTR, Config, DataLoader};
#[derive(argh::FromArgs)]
/// Example
@@ -22,7 +22,7 @@ fn main() -> Result<()> {
let args: Args = argh::from_env();
// build model
let config = ModelConfig::ppocr_rec_v4_ch()
let config = Config::ppocr_rec_v4_ch()
// ppocr_rec_v4_en()
// repsvtr_ch()
.with_model_device(args.device.as_str().try_into()?)

10
examples/trocr/main.rs
View File

@@ -1,6 +1,6 @@
use usls::{
models::{TrOCR, TrOCRKind},
DataLoader, ModelConfig, Scale,
Config, DataLoader, Scale,
};
#[derive(argh::FromArgs)]
@@ -40,12 +40,12 @@ fn main() -> anyhow::Result<()> {
// build model
let config = match args.scale.as_str().try_into()? {
Scale::S => match args.kind.as_str().try_into()? {
TrOCRKind::Printed => ModelConfig::trocr_small_printed(),
TrOCRKind::HandWritten => ModelConfig::trocr_small_handwritten(),
TrOCRKind::Printed => Config::trocr_small_printed(),
TrOCRKind::HandWritten => Config::trocr_small_handwritten(),
},
Scale::B => match args.kind.as_str().try_into()? {
TrOCRKind::Printed => ModelConfig::trocr_base_printed(),
TrOCRKind::HandWritten => ModelConfig::trocr_base_handwritten(),
TrOCRKind::Printed => Config::trocr_base_printed(),
TrOCRKind::HandWritten => Config::trocr_base_handwritten(),
},
x => anyhow::bail!("Unsupported TrOCR scale: {:?}", x),
}

4
examples/ultralytics-rtdetr/main.rs
View File

@@ -1,5 +1,5 @@
use anyhow::Result;
use usls::{models::YOLO, Annotator, DataLoader, ModelConfig};
use usls::{models::YOLO, Annotator, Config, DataLoader};
#[derive(argh::FromArgs)]
/// Example
@@ -22,7 +22,7 @@ fn main() -> Result<()> {
let args: Args = argh::from_env();
// build model
let config = ModelConfig::ultralytics_rtdetr_l()
let config = Config::ultralytics_rtdetr_l()
.with_model_dtype(args.dtype.as_str().try_into()?)
.with_model_device(args.device.as_str().try_into()?)
.commit()?;

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

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

9
examples/yolo/main.rs
View File

@@ -1,7 +1,7 @@
use anyhow::Result;
use usls::{
models::YOLO, Annotator, DataLoader, ModelConfig, Style, NAMES_COCO_80,
NAMES_COCO_KEYPOINTS_17, NAMES_IMAGENET_1K, SKELETON_COCO_19, SKELETON_COLOR_COCO_19,
models::YOLO, Annotator, Config, DataLoader, Style, NAMES_COCO_80, NAMES_COCO_KEYPOINTS_17,
NAMES_IMAGENET_1K, SKELETON_COCO_19, SKELETON_COLOR_COCO_19,
};
#[derive(argh::FromArgs, Debug)]
@@ -130,7 +130,7 @@ fn main() -> Result<()> {
.with_timer(tracing_subscriber::fmt::time::ChronoLocal::rfc_3339())
.init();
let args: Args = argh::from_env();
let mut config = ModelConfig::yolo()
let mut config = Config::yolo()
.with_model_file(&args.model.unwrap_or_default())
.with_task(args.task.as_str().try_into()?)
.with_version(args.ver.try_into()?)
@@ -170,7 +170,8 @@ fn main() -> Result<()> {
})
.with_topk(args.topk)
.retain_classes(&args.retain_classes)
.exclude_classes(&args.exclude_classes);
.exclude_classes(&args.exclude_classes)
.with_model_num_dry_run(2);
if args.use_coco_80_classes {
config = config.with_class_names(&NAMES_COCO_80);
}

4
examples/yoloe/main.rs
View File

@@ -1,5 +1,5 @@
use anyhow::Result;
use usls::{models::YOLO, Annotator, DataLoader, ModelConfig, Style};
use usls::{models::YOLO, Annotator, Config, DataLoader, Style};
#[derive(argh::FromArgs)]
/// Example
@@ -22,7 +22,7 @@ fn main() -> Result<()> {
let args: Args = argh::from_env();
// config
let config = ModelConfig::yoloe_v8s_seg_pf()
let config = Config::yoloe_v8s_seg_pf()
// yoloe_v8m_seg_pf()
// yoloe_v8l_seg_pf()
// yoloe_11s_seg_pf()

4
examples/yolop/main.rs
View File

@@ -1,5 +1,5 @@
use anyhow::Result;
use usls::{models::YOLOPv2, Annotator, DataLoader, ModelConfig};
use usls::{models::YOLOPv2, Annotator, Config, DataLoader};
fn main() -> Result<()> {
tracing_subscriber::fmt()
@@ -8,7 +8,7 @@ fn main() -> Result<()> {
.init();
// build model
let mut model = YOLOPv2::new(ModelConfig::yolop_v2_480x800().commit()?)?;
let mut model = YOLOPv2::new(Config::yolop_v2_480x800().commit()?)?;
// load image
let xs = DataLoader::try_read_n(&["images/car-view.jpg"])?;