mirror of
https://github.com/mii443/usls.git
synced 2025-08-22 15:45:41 +00:00
Update ort and improve the speed of preprocessing
* Add onnx proto * Update ort to 2.0.0-rc.2 * Improve the speed of resizing * Fix yolo-seg bug * Update README.md
This commit is contained in:
@ -4,9 +4,9 @@ fn main() -> Result<(), Box<dyn std::error::Error>> {
|
||||
// build model
|
||||
let options = Options::default()
|
||||
.with_model("rtdetr-l-f16.onnx")?
|
||||
.with_confs(&[0.4, 0.15]) // person: 0.4, others: 0.15
|
||||
.with_confs(&[0.4, 0.15])
|
||||
.with_names(&coco::NAMES_80);
|
||||
let mut model = RTDETR::new(&options)?;
|
||||
let mut model = RTDETR::new(options)?;
|
||||
|
||||
// load image
|
||||
let x = vec![DataLoader::try_read("./assets/bus.jpg")?];
|
||||
|
Reference in New Issue
Block a user