Add onnx proto

jamjamjon · May 7, 2024 · b7615b3 · b7615b3
1 parent 246e609
commit b7615b3
Show file tree

Hide file tree

Showing 6 changed files with 1,335 additions and 49 deletions.
diff --git a/Cargo.toml b/Cargo.toml
@@ -41,3 +41,6 @@ image = "0.25.1"
 imageproc = { version = "0.24" }
 ab_glyph = "0.2.23"
 geo = "0.28.0"
+prost = "0.12.4"
+human_bytes = "0.4.3"
+sysinfo = "0.30.12"
diff --git a/examples/yolov8/main.rs b/examples/yolov8/main.rs
@@ -9,8 +9,9 @@ fn main() -> Result<(), Box<dyn std::error::Error>> {
         // .with_model("yolov8m-seg-dyn.onnx")?
         // .with_model("yolov8m-obb-dyn.onnx")?
         // .with_model("yolov8m-oiv7-dyn.onnx")?
-        // .with_trt(0)
+        .with_trt(0)
         // .with_fp16(true)
+        // .with_coreml(0)
         .with_i00((1, 1, 4).into())
         .with_i02((224, 640, 800).into())
         .with_i03((224, 640, 800).into())