diff --git a/polygon-yolov5/requirements.txt b/polygon-yolov5/requirements.txt index ee901d4..347026b 100644 --- a/polygon-yolov5/requirements.txt +++ b/polygon-yolov5/requirements.txt @@ -7,7 +7,7 @@ opencv-python>=4.1.2 Pillow PyYAML>=5.3.1 scipy>=1.4.1 -torch>=1.7.0 +torch>=2.2.0 torchvision>=0.8.1 tqdm>=4.41.0 shapely