ultralytics-rust-engine 0.1.0

A complete pure-Rust reimplementation of Ultralytics YOLO: train, val, predict and export for the full YOLO family (v3/v5/v8/v9/v10/v11/v12/v26), zero Python dependency.
1
2
3
4
5
6
7
8
9
10
11
12
//! ul-engine — 引擎层:损失、训练、验证、预测、导出。

pub mod engine;
pub mod loss;
pub mod nms;

pub use engine::{
    load_checkpoint, predict, save_checkpoint, train, validate_model, validate_task, ModelEma,
    PredictOut, TrainCfg, TrainReport,
};
pub use loss::{classification_loss, v8_detection_loss, DetBatch, LossW};
pub use nms::{non_max_suppression_ex, NmsCfg};