diff --git a/README.md b/README.md index c9166862..fb19abfa 100755 --- a/README.md +++ b/README.md @@ -26,12 +26,7 @@ The https://github.com/ultralytics/yolov3 repo contains inference and training c # Requirements -Python 3.7 or later with all of the `pip install -U -r requirements.txt` packages including: -- `torch >= 1.4` -- `opencv-python` -- `Pillow` - -All dependencies are included in the associated docker images. Docker requirements are: +Python 3.7 or later with all of the `pip install -U -r requirements.txt` packages including `torch >= 1.5`. Docker images come with all dependencies preinstalled. Docker requirements are: - Nvidia Driver >= 440.44 - Docker Engine - CE >= 19.03 diff --git a/requirements.txt b/requirements.txt index a4684389..e705c8c0 100755 --- a/requirements.txt +++ b/requirements.txt @@ -1,7 +1,7 @@ # pip install -U -r requirements.txt numpy opencv-python >= 4.1 -torch >= 1.4 +torch >= 1.5 matplotlib pycocotools tqdm