Update requirements.txt

requirements.txt

@@ -1,144 +1,49 @@
-absl-py==2.1.0
-addict==2.4.0
-aiofiles==23.2.1
-albucore==0.0.17
-albumentations==1.4.18
-aliyun-python-sdk-core==2.15.1
-aliyun-python-sdk-kms==2.16.3
-annotated-types==0.7.0
-anyio==4.5.2
-cachetools==5.5.0
-certifi==2024.6.2
-charset-normalizer==3.3.2
-click==8.1.8
-contourpy==1.1.1
-crcmod==1.7
--e git+https://huggingface.co/spaces/yuhangzang/ContextDet-Demo@7be5aa59f3ba3f5bf38039f6ce80df64913c5627#egg=csrc
-cycler==0.12.1
-Cython==3.0.10
-et-xmlfile==1.1.0
-eval_type_backport==0.2.2
-exceptiongroup==1.2.2
-fastapi==0.115.11
-ffmpy==0.5.0
-filelock==3.14.0
-fonttools==4.53.1
-fsspec==2024.6.1
-future==1.0.0
-gitdb==4.0.11
-GitPython==3.1.43
-google-auth==2.37.0
-google-auth-oauthlib==1.0.0
-gradio==4.44.1
-gradio_client==1.3.0
-grpcio==1.69.0
-h11==0.14.0
-h5py==3.11.0
-httpcore==1.0.7
-httpx==0.28.1
-huggingface-hub==0.29.3
-idna==3.7
-imageio==2.35.1
-importlib_metadata==8.2.0
-importlib_resources==6.4.0
-inplace-abn==1.1.0
-Jinja2==3.1.4
-jmespath==0.10.0
-kiwisolver==1.4.5
-lazy_loader==0.4
-Markdown==3.6
-markdown-it-py==3.0.0
-MarkupSafe==2.1.5
-matplotlib==3.7.5
-mdurl==0.1.2
-mmengine==0.10.4
-model-index==0.1.11
-mpmath==1.3.0
-networkx==3.1
-numpy==1.24.4
-nvidia-cublas-cu12==12.1.3.1
-nvidia-cuda-cupti-cu12==12.1.105
-nvidia-cuda-nvrtc-cu12==12.1.105
-nvidia-cuda-runtime-cu12==12.1.105
-nvidia-cudnn-cu12==8.9.2.26
-nvidia-cufft-cu12==11.0.2.54
-nvidia-curand-cu12==10.3.2.106
-nvidia-cusolver-cu12==11.4.5.107
-nvidia-cusparse-cu12==12.1.0.106
-nvidia-nccl-cu12==2.20.5
-nvidia-nvjitlink-cu12==12.5.82
-nvidia-nvtx-cu12==12.1.105
-oauthlib==3.2.2
-opencv-python==4.10.0.84
-opencv-python-headless==4.11.0.86
-opendatalab==0.0.10
-openpyxl==3.1.5
-openxlab==0.1.1
-ordered-set==4.1.0
-orjson==3.10.15
-oss2==2.17.0
-packaging==24.1
-pandas==2.0.3
-pillow==10.4.0
-platformdirs==4.2.2
-protobuf==5.27.2
-psutil==6.0.0
-py-cpuinfo==9.0.0
-pyasn1==0.6.1
-pyasn1_modules==0.4.1
-pycocotools==2.0.7
-pycryptodome==3.20.0
-pydantic==2.10.5
-pydantic_core==2.27.2
-pydub==0.25.1
-Pygments==2.18.0
-pyparsing==3.1.2
-python-dateutil==2.9.0.post0
-python-multipart==0.0.20
-pytz==2023.4
-PyWavelets==1.4.1
-PyYAML==6.0.1
-randaugment==1.0.2
-requests==2.32.3
-requests-oauthlib==2.0.0
-rich==13.4.2
-rsa==4.9
-ruff==0.11.0
-salesforce-lavis==1.0.2
-scikit-image==0.21.0
-scipy==1.10.1
-seaborn==0.13.2
-semantic-version==2.10.0
-shapely==2.0.5
-shellingham==1.5.4
-six==1.16.0
-smmap==5.0.1
-sniffio==1.3.1
-starlette==0.44.0
-sympy==1.13.0
-tabulate==0.9.0
-tensorboard==2.14.0
-tensorboard-data-server==0.7.2
-tensorboardX==2.6.2.2
-termcolor==2.4.0
-terminaltables==3.1.10
-thop==0.1.1.post2209072238
-tifffile==2023.7.10
-tomli==2.0.1
-tomlkit==0.12.0
-torch==2.3.1
-torchaudio==0.4.0
-torchvision==0.18.1
-tqdm==4.65.2
-triton==2.3.1
-typer==0.15.2
-typing_extensions==4.12.2
-tzdata==2024.1
-ultralytics==8.2.66
-ultralytics-thop==2.0.0
-urllib3==2.2.2
-uvicorn==0.33.0
-webdataset==0.2.100
-websockets==12.0
-Werkzeug==3.0.6
-zipp==3.19.2
+# YOLOv5 requirements
+# Usage: pip install -r requirements.txt
+
+# Base ------------------------------------------------------------------------
+gitpython>=3.1.30
+matplotlib>=3.3
+numpy>=1.23.5
+opencv-python>=4.1.1
+pillow>=10.3.0
+psutil  # system resources
+PyYAML>=5.3.1
+requests>=2.32.2
+scipy>=1.4.1
+thop>=0.1.1  # FLOPs computation
+torch>=1.8.0  # see https://pytorch.org/get-started/locally (recommended)
+torchvision>=0.9.0
+tqdm>=4.66.3
+ultralytics>=8.2.34  # https://ultralytics.com
+# protobuf<=3.20.1  # https://github.com/ultralytics/yolov5/issues/8012
+
+# Logging ---------------------------------------------------------------------
+# tensorboard>=2.4.1
+# clearml>=1.2.0
+# comet
+
+# Plotting --------------------------------------------------------------------
+pandas>=1.1.4
+seaborn>=0.11.0
+
+# Export ----------------------------------------------------------------------
+# coremltools>=6.0  # CoreML export
+# onnx>=1.10.0  # ONNX export
+# onnx-simplifier>=0.4.1  # ONNX simplifier
+# nvidia-pyindex  # TensorRT export
+# nvidia-tensorrt  # TensorRT export
+# scikit-learn<=1.1.2  # CoreML quantization
+# tensorflow>=2.4.0,<=2.13.1  # TF exports (-cpu, -aarch64, -macos)
+# tensorflowjs>=3.9.0  # TF.js export
+# openvino-dev>=2023.0  # OpenVINO export
+
+# Deploy ----------------------------------------------------------------------
+setuptools>=70.0.0 # Snyk vulnerability fix
+# tritonclient[all]~=2.24.0
+
+# Extras ----------------------------------------------------------------------
+# ipython  # interactive notebook
+# mss  # screenshots
+# albumentations>=1.0.3
+# pycocotools>=2.0.6  # COCO mAP