Add YOLOv11

* Add YOLOv11
jamjamjon · Sep 30, 2024 · 0609dd1 · 0609dd1
1 parent 2cb9e57
commit 0609dd1
Show file tree

Hide file tree

Showing 5 changed files with 299 additions and 199 deletions.
diff --git a/Cargo.toml b/Cargo.toml
@@ -1,6 +1,6 @@
 [package]
 name = "usls"
-version = "0.0.16"
+version = "0.0.17"
 edition = "2021"
 description = "A Rust library integrated with ONNXRuntime, providing a collection of ML models."
 repository = "https://github.com/jamjamjon/usls"

diff --git a/examples/yolo/README.md b/examples/yolo/README.md
@@ -24,45 +24,41 @@
 ## Quick Start
 ```Shell
 
+# customized
+cargo run -r --example yolo -- --task detect --ver v8 --nc 6 --model xxx.onnx  # YOLOv8
+
 # Classify
-cargo run -r --example yolo -- --task classify --ver v5  # YOLOv5 
-cargo run -r --example yolo -- --task classify --ver v8  # YOLOv8 
+cargo run -r --example yolo -- --task classify --ver v5 --scale s --width 224 --height 224 --nc 1000  # YOLOv5
+cargo run -r --example yolo -- --task classify --ver v8 --scale n --width 224 --height 224 --nc 1000  # YOLOv8 
+cargo run -r --example yolo -- --task classify --ver v11 --scale n --width 224 --height 224 --nc 1000  # YOLOv11 
 
 # Detect
-cargo run -r --example yolo -- --task detect --ver v5  # YOLOv5 
-cargo run -r --example yolo -- --task detect --ver v6  # YOLOv6
-cargo run -r --example yolo -- --task detect --ver v7  # YOLOv7
-cargo run -r --example yolo -- --task detect --ver v8  # YOLOv8
-cargo run -r --example yolo -- --task detect --ver v9  # YOLOv9
-cargo run -r --example yolo -- --task detect --ver v10 # YOLOv10
-cargo run -r --example yolo -- --task detect --ver rtdetr  # YOLOv8-RTDETR
-cargo run -r --example yolo -- --task detect --ver v8 --model yolov8s-world-v2-shoes.onnx  # YOLOv8-world
+cargo run -r --example yolo -- --task detect --ver v5 --scale n  # YOLOv5 
+cargo run -r --example yolo -- --task detect --ver v6 --scale n  # YOLOv6
+cargo run -r --example yolo -- --task detect --ver v7 --scale t  # YOLOv7
+cargo run -r --example yolo -- --task detect --ver v8 --scale n  # YOLOv8
+cargo run -r --example yolo -- --task detect --ver v9 --scale t  # YOLOv9
+cargo run -r --example yolo -- --task detect --ver v10 --scale n  # YOLOv10
+cargo run -r --example yolo -- --task detect --ver v11 --scale n  # YOLOv11
+cargo run -r --example yolo -- --task detect --ver rtdetr --scale l  # RTDETR
+cargo run -r --example yolo -- --task detect --ver v8 --nc 1 --model yolov8s-world-v2-shoes.onnx  # YOLOv8-world <local file>
 
 # Pose
-cargo run -r --example yolo -- --task pose --ver v8  # YOLOv8-Pose
+cargo run -r --example yolo -- --task pose --ver v8 --scale n   # YOLOv8-Pose
+cargo run -r --example yolo -- --task pose --ver v11 --scale n  # YOLOv11-Pose
 
 # Segment
-cargo run -r --example yolo -- --task segment --ver v5  # YOLOv5-Segment
-cargo run -r --example yolo -- --task segment --ver v8  # YOLOv8-Segment
-cargo run -r --example yolo -- --task segment --ver v8 --model FastSAM-s-dyn-f16.onnx  # FastSAM
+cargo run -r --example yolo -- --task segment --ver v5 --scale n  # YOLOv5-Segment
+cargo run -r --example yolo -- --task segment --ver v8 --scale n  # YOLOv8-Segment
+cargo run -r --example yolo -- --task segment --ver v11 --scale n  # YOLOv8-Segment
+cargo run -r --example yolo -- --task segment --ver v8 --model FastSAM-s-dyn-f16.onnx  # FastSAM <local file>
 
 # Obb
-cargo run -r --example yolo -- --task obb --ver v8  # YOLOv8-Obb
+cargo run -r --example yolo -- --ver v8 --task obb --scale n --width 1024 --height 1024 --source images/dota.png  # YOLOv8-Obb
+cargo run -r --example yolo -- --ver v11 --task obb --scale n --width 1024 --height 1024 --source images/dota.png  # YOLOv11-Obb
 ```
 
-<details close>
-<summary>other options</summary>
-
-`--source` to specify the input images  
-`--model` to specify the ONNX model  
-`--width --height` to specify the input resolution  
-`--nc` to specify the number of model's classes  
-`--plot` to annotate with inference results  
-`--profile` to profile  
-`--cuda --trt --coreml --device_id` to select device  
-`--half` to use float16 when using TensorRT EP  
-
-</details>
+**`cargo run -r --example yolo -- --help` for more options**
 
 
 ## YOLOs configs with `Options` 
@@ -74,6 +70,8 @@ cargo run -r --example yolo -- --task obb --ver v8  # YOLOv8-Obb
 let options = Options::default()
     .with_yolo_version(YOLOVersion::V5)  // YOLOVersion: V5, V6, V7, V8, V9, V10, RTDETR
     .with_yolo_task(YOLOTask::Classify)  // YOLOTask: Classify, Detect, Pose, Segment, Obb
+    // .with_nc(80)
+    // .with_names(&COCO_CLASS_NAMES_80)
     .with_model("xxxx.onnx")?;
 
 ```
@@ -140,7 +138,7 @@ let options = Options::default()
 </details>
 
 <details close>
-<summary>YOLOv8</summary>
+<summary>YOLOv8, YOLOv11</summary>
 
 ```Shell
 pip install -U ultralytics