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::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"])?;