Add onnx

baernhaeckt · Aug 25, 2024 · 09959df · 09959df
1 parent 5790234
commit 09959df
Show file tree

Hide file tree

Showing 2 changed files with 177 additions and 1 deletion.
diff --git a/poetry.lock b/poetry.lock
diff --git a/pyproject.toml b/pyproject.toml
@@ -12,6 +12,9 @@ uvicorn = "^0.30.6"
 pydantic-settings = "^2.4.0"
 ultralytics = "^8.2.81"
 python-multipart = "^0.0.9"
+onnx = "^1.16.2"
+onnxruntime = "^1.19.0"
+onnxslim = "^0.1.32"
 
 
 [tool.poetry.group.dev.dependencies]