mirror of
https://github.com/mii443/usls.git
synced 2025-08-22 15:45:41 +00:00
Add some eps (#108)
This commit is contained in:
@ -47,9 +47,9 @@ fn main() -> Result<()> {
|
||||
// build model
|
||||
let config = match &args.model {
|
||||
Some(m) => Config::db().with_model_file(m),
|
||||
None => Config::ppocr_det_v5_mobile().with_model_dtype(args.dtype.as_str().try_into()?),
|
||||
None => Config::ppocr_det_v5_mobile().with_model_dtype(args.dtype.parse()?),
|
||||
}
|
||||
.with_device_all(args.device.as_str().try_into()?)
|
||||
.with_device_all(args.device.parse()?)
|
||||
.commit()?;
|
||||
let mut model = DB::new(config)?;
|
||||
|
||||
|
Reference in New Issue
Block a user