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

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"),
};