| # Deep Learning Framework ------------------------------------- | |
| torch>=1.8.0 | |
| torchvision>=0.9.0 | |
| # Web Interface and Visualization ---------------------------- | |
| gradio>=4.0.0 | |
| matplotlib>=3.3 | |
| seaborn>=0.11.0 | |
| pandas>=1.1.4 | |
| # Image Processing and Data Handling ------------------------- | |
| opencv-python>=4.1.1 | |
| Pillow>=7.1.2 | |
| numpy>=1.22.2 | |
| scipy>=1.4.1 | |
| requests>=2.23.0 | |
| # YOLOv5 Specific Dependencies ------------------------------ | |
| ultralytics>=8.0.147 | |
| thop>=0.1.1 # FLOPs computation | |
| PyYAML>=5.3.1 | |
| psutil # system resources | |
| tqdm>=4.64.0 | |
| gitpython>=3.1.30 | |
| # CUDA Support (optional) ----------------------------------- | |
| --extra-index-url https://download.pytorch.org/whl/cu118 | |
| torch | |
| torchvision | |
| # Development Tools (optional) ------------------------------ | |
| # tensorboard>=2.4.1 | |
| # clearml>=1.2.0 | |
| # ipython # interactive notebook | |
| # albumentations>=1.0.3 | |
| # pycocotools>=2.0.6 # COCO mAP | |
| # Export Tools (optional) ---------------------------------- | |
| # onnx>=1.10.0 # ONNX export | |
| # onnx-simplifier>=0.4.1 # ONNX simplifier | |
| # coremltools>=6.0 # CoreML export | |
| # tensorflow>=2.4.0 # TF exports | |
| # tensorflowjs>=3.9.0 # TF.js export | |
| # openvino-dev>=2023.0 # OpenVINO export |