feat(dependencies): add ONNX and ONNX Runtime support; downgrade Pyth…

…on version to 3.10
hotosm · Nov 23, 2024 · 17382fc · 17382fc
1 parent 75b1633
commit 17382fc
Show file tree

Hide file tree

Showing 4 changed files with 6 additions and 4 deletions.
diff --git a/.github/workflows/unit-test.yml b/.github/workflows/unit-test.yml
@@ -20,7 +20,7 @@ jobs:
       - name: Install dependencies
         run: |
           python -m pip install --upgrade pip
-          pip install tensorflow==2.12.0 efficientnet==1.1.1 torch==2.4.1 ultralytics==8.3.26 tflite-runtime==2.12.0
+          pip install tensorflow==2.12.0 efficientnet==1.1.1 torch==2.4.1 ultralytics==8.3.26 onnx==1.17.0 onnxruntime==1.20.1
           pip install -e .
 
       - name: Run tests

diff --git a/API/Dockerfile b/API/Dockerfile
@@ -1,4 +1,4 @@
-ARG PYTHON_VERSION=3.11
+ARG PYTHON_VERSION=3.10
 
 FROM docker.io/python:${PYTHON_VERSION}-slim-bookworm
 

diff --git a/API/requirements.txt b/API/requirements.txt
@@ -4,4 +4,6 @@ fairpredictor
 # tflite-runtime==2.14.0
 tensorflow==2.12.0 
 efficientnet==1.1.1
-hot-fair-utilities==2.0.2 ## TODO : Get rid of this after the testing with YOLOv1
+onnx==1.17.0
+onnxruntime==1.20.1
+# hot-fair-utilities==2.0.2 ## TODO : Get rid of this after the testing with YOLOv1
diff --git a/setup.py b/setup.py
@@ -34,6 +34,6 @@
         "shapely>=1.0.0,<=2.0.2",
         "rasterio>=1.0.0,<=1.3.8",
         "orthogonalizer<=0.0.4",
-        "hot-fair-utilities==2.0.2",
+        # "hot-fair-utilities==2.0.2",
     ],
 )