chore(deps): models-yolox: -r requirements.txt

Author: YodaEmbedding

pyproject.toml

@@ -74,6 +74,18 @@ models-mmpose = [
     "scipy",
     "xtcocotools>=1.12",
 ]
+models-yolox = [
+    "loguru",
+    "ninja",
+    # "onnx-simplifier==0.4.10",
+    # "onnx>=1.13.0",
+    "psutil",
+    "pycocotools>=2.0.7",
+    "tabulate",
+    # "tensorboard",
+    "thop",
+    "tqdm",
+]
 
 # NOTE: Temporarily duplicated from [project.optional-dependencies] until
 # pip supports installing [dependency-groups].
@@ -111,6 +123,18 @@ models-mmpose = [
     "scipy",
     "xtcocotools>=1.12",
 ]
+models-yolox = [
+    "loguru",
+    "ninja",
+    # "onnx-simplifier==0.4.10",
+    # "onnx>=1.13.0",
+    "psutil",
+    "pycocotools>=2.0.7",
+    "tabulate",
+    # "tensorboard",
+    "thop",
+    "tqdm",
+]
 
 [project.scripts]
 "compressai-vision-eval" = "compressai_vision.run.eval_split_inference:main"

scripts/install_uv.sh

@@ -221,6 +221,8 @@ install_yolox () {
     echo "Installing YOLOX (reference: https://github.com/Megvii-BaseDetection/YOLOX)"
     echo
 
+    uv sync --inexact --group=models-yolox
+
     # clone
     if [ -z "$(ls -A ${MODELS_SOURCE_DIR}/yolox)" ]; then
         git clone https://github.com/Megvii-BaseDetection/yolox.git ${MODELS_SOURCE_DIR}/yolox
@@ -232,8 +234,6 @@ install_yolox () {
     fi
 
     cd ${MODELS_SOURCE_DIR}/yolox
-    # miminum requirments - no onnx, etc.
-    cp ${SCRIPT_DIR}/yolox_requirements.txt requirements.txt
 
     uv pip install --no-build-isolation .