入门指南

此库依赖于tch crate，用于绑定C++ Libtorch API。有关此库的详细安装信息，请参阅此仓库链接。

推理

use rusty_yolo;
use tch;

fn main() {
    let device = tch::Device::cuda_if_available();
    let yolo_model = rusty_yolo::YOLO::new("model.pt", 384, 640, device);
    let mut original_image = tch::vision::image::load("/images/zidane.jpg").unwrap();

    let results = yolo_model.predict(&original_image, 0.25, 0.35);
    yolo_model.draw_rectangle(&mut original_image, &results);
    tch::vision::image::save(&original_image, "images/result.jpg").unwrap();
}