diff --git a/requirements.txt b/requirements.txt index 6c7489611..54cac7a2e 100644 --- a/requirements.txt +++ b/requirements.txt @@ -12,8 +12,8 @@ PyYAML>=5.3.1 requests>=2.23.0 scipy>=1.4.1 thop>=0.1.1 # FLOPs computation -torch>=1.7.0,<2 # see https://pytorch.org/get-started/locally (recommended) -torchvision>=0.8.1,<0.15 +torch>=1.10.1,<2 +torchvision>=0.11,<0.15 tqdm>=4.64.0 # protobuf<=3.20.1 # https://github.com/ultralytics/yolov5/issues/8012 @@ -36,7 +36,7 @@ onnx-simplifier>=0.4.1 # ONNX simplifier # tensorflow>=2.4.1 # TF exports (-cpu, -aarch64, -macos) # tensorflowjs>=3.9.0 # TF.js export # openvino-dev # OpenVINO export -rknn-toolkit2 +rknn-toolkit2==2.3 # Deploy ---------------------------------------------------------------------- setuptools>=65.5.1 # Snyk vulnerability fix