Use ML models package and remove custom ml functions

dnum-mi · Jun 12, 2024 · d30a161 · d30a161
1 parent 24f1453
commit d30a161
Show file tree

Hide file tree

Showing 10 changed files with 23 additions and 420 deletions.
diff --git a/backend/requirements.txt b/backend/requirements.txt
@@ -10,9 +10,7 @@ user-agents==2.2.0
 boto3==1.28.39
 autodynatrace==2.0.0
 # ML
-ultralytics==8.1.2
-opencv-python==4.9.0.80
-onnxruntime==1.17.1
+https://github.com/dnum-mi/basegun-ml/raw/MLPackages/MLpackages/basegun_ml/dist/basegunml-0.1.tar.gz
 # Dev
 pytest==7.4.3
 coverage==7.3.2
diff --git a/backend/src/ml/measure/best_card.onnx b/backend/src/ml/measure/best_card.onnx
diff --git a/backend/src/ml/measure/best_keypoints.pt b/backend/src/ml/measure/best_keypoints.pt
diff --git a/backend/src/ml/measure/measure.py b/backend/src/ml/measure/measure.py
diff --git a/backend/src/ml/measure/warmup.jpg b/backend/src/ml/measure/warmup.jpg
diff --git a/backend/src/ml/models/typology.pt b/backend/src/ml/models/typology.pt