Update PyTorch and dependency versions in pyproject.toml and README

- Add Python version-specific PyTorch and torchvision dependencies
- Update MMCV find-links to torch2.1.0
- Remove Detectron2 and Super-Gradients installation instructions
- Update package versions for Ultralytics and YOLOv5
- Simplify installation instructions for different frameworks

Author: fcakyon

README.md

@@ -97,8 +97,6 @@ Object detection and instance segmentation are by far the most important applica
 
 - `MMDetection` + `SAHI` walkthrough: <a href="https://colab.research.google.com/github/obss/sahi/blob/main/demo/inference_for_mmdetection.ipynb"><img src="https://colab.research.google.com/assets/colab-badge.svg" alt="sahi-mmdetection"></a>
 
-- `Detectron2` + `SAHI` walkthrough: <a href="https://colab.research.google.com/github/obss/sahi/blob/main/demo/inference_for_detectron2.ipynb"><img src="https://colab.research.google.com/assets/colab-badge.svg" alt="sahi-detectron2"></a>
-
 - `TorchVision` + `SAHI` walkthrough: <a href="https://colab.research.google.com/github/obss/sahi/blob/main/demo/inference_for_torchvision.ipynb"><img src="https://colab.research.google.com/assets/colab-badge.svg" alt="sahi-torchvision"></a>
 
 <a href="https://huggingface.co/spaces/fcakyon/sahi-yolox"><img width="600" src="https://user-images.githubusercontent.com/34196005/144092739-c1d9bade-a128-4346-947f-424ce00e5c4f.gif" alt="sahi-yolox"></a>
@@ -126,39 +124,35 @@ pip install sahi
 conda install -c conda-forge shapely
 ```
 
-- Install your desired version of pytorch and torchvision (cuda 11.3 for detectron2, cuda 11.7 for rest):
+- Install your desired version of pytorch and torchvision:
 
 ```console
-conda install pytorch=1.10.2 torchvision=0.11.3 cudatoolkit=11.3 -c pytorch
+pip install torch==2.6.0 torchvision==0.21.0 --index-url https://download.pytorch.org/whl/cu126
 ```
 
+(torch 2.1.2 is required for mmdet support):
+
 ```console
-conda install pytorch=1.13.1 torchvision=0.14.1 pytorch-cuda=11.7 -c pytorch -c nvidia
+pip install torch==2.1.2 torchvision==0.16.2 --index-url https://download.pytorch.org/whl/cu121
 ```
 
 - Install your desired detection framework (yolov5):
 
 ```console
-pip install yolov5==7.0.13
+pip install yolov5==7.0.14
 ```
 
 - Install your desired detection framework (ultralytics):
 
 ```console
-pip install ultralytics==8.3.50
+pip install ultralytics==8.3.86
 ```
 
 - Install your desired detection framework (mmdet):
 
 ```console
 pip install mim
-mim install mmdet==3.0.0
-```
-
-- Install your desired detection framework (detectron2):
-
-```console
-pip install detectron2 -f https://dl.fbaipublicfiles.com/detectron2/wheels/cu113/torch1.10/index.html
+mim install mmdet==3.3.0
 ```
 
 - Install your desired detection framework (huggingface):
@@ -167,12 +161,6 @@ pip install detectron2 -f https://dl.fbaipublicfiles.com/detectron2/wheels/cu113
 pip install transformers timm
 ```
 
-- Install your desired detection framework (super-gradients):
-
-```console
-pip install super-gradients==3.3.1
-```
-
 </details>
 
 ### Framework Agnostic Sliced/Standard Prediction

pyproject.toml

@@ -41,7 +41,7 @@ sahi = "sahi.cli:app"
 
 [tool.uv]
 find-links = [
-  "https://download.openmmlab.com/mmcv/dist/cpu/torch2.3.0/index.html"
+  "https://download.openmmlab.com/mmcv/dist/cpu/torch2.1.0/index.html"
 ]
 default-groups = ["dev", "ci"]
 
@@ -55,8 +55,11 @@ dev = [
 ]
 ci = [
   # pytorch should be present for all python versions
-  "torch==2.3.1+cpu",
-  "torchvision==0.18.1+cpu",
+  "torch==2.6.0+cpu;python_version>='3.12'",
+  "torchvision==0.21.0+cpu;python_version>='3.12'",
+  # mmdet is supported for python < 3.12
+  "torch==2.1.2+cpu;python_version<'3.12'",
+  "torchvision==0.16.2+cpu;python_version<'3.12'",
   "mmengine;python_version<'3.12'",
   "mmcv==2.1.0;python_version<'3.12'",
   "mmdet==3.3.0;python_version<'3.12'",