From 53f95e5a5fa65c8d191ed80d18f881933a45b903 Mon Sep 17 00:00:00 2001 From: Shriya Balaji Palsamudram Date: Thu, 4 Nov 2021 14:13:48 -0700 Subject: [PATCH] [MRCNN-PYT] Update opencv-python version, fix numpy version --- PyTorch/Segmentation/MaskRCNN/Dockerfile | 4 ++-- PyTorch/Segmentation/MaskRCNN/README.md | 2 +- PyTorch/Segmentation/MaskRCNN/pytorch/Dockerfile | 4 ++-- PyTorch/Segmentation/MaskRCNN/pytorch/requirements.txt | 2 +- 4 files changed, 6 insertions(+), 6 deletions(-) diff --git a/PyTorch/Segmentation/MaskRCNN/Dockerfile b/PyTorch/Segmentation/MaskRCNN/Dockerfile index 27d19b1c..0b97e7ad 100644 --- a/PyTorch/Segmentation/MaskRCNN/Dockerfile +++ b/PyTorch/Segmentation/MaskRCNN/Dockerfile @@ -18,7 +18,7 @@ FROM ${FROM_IMAGE_NAME} # Install Python dependencies RUN pip install --no-cache-dir \ mlperf-compliance==0.0.10 \ - opencv-python==4.2.0.32 \ + opencv-python==4.4.0.42 \ git+https://github.com/NVIDIA/dllogger \ yacs @@ -29,6 +29,6 @@ RUN mkdir /results COPY pytorch/. . RUN pip install -e . -RUN /opt/conda/bin/conda install -y numpy==1.17.4 +RUN /opt/conda/bin/conda install -y numpy ENV OMP_NUM_THREADS=1 diff --git a/PyTorch/Segmentation/MaskRCNN/README.md b/PyTorch/Segmentation/MaskRCNN/README.md index 796def46..bf1207c2 100755 --- a/PyTorch/Segmentation/MaskRCNN/README.md +++ b/PyTorch/Segmentation/MaskRCNN/README.md @@ -597,7 +597,7 @@ To achieve these same results, follow the steps in the [Quick Start Guide](#quic October 2021 - Replace APEX AMP with PyTorch native AMP -- Use opencv-python version 4.2.0.32 +- Use opencv-python version 4.4.0.42 June 2020 - Updated accuracy and performance tables to include A100 results diff --git a/PyTorch/Segmentation/MaskRCNN/pytorch/Dockerfile b/PyTorch/Segmentation/MaskRCNN/pytorch/Dockerfile index 2f686ac4..c9d75ba2 100755 --- a/PyTorch/Segmentation/MaskRCNN/pytorch/Dockerfile +++ b/PyTorch/Segmentation/MaskRCNN/pytorch/Dockerfile @@ -18,7 +18,7 @@ FROM ${FROM_IMAGE_NAME} # Install Python dependencies RUN pip install --no-cache-dir \ mlperf-compliance==0.0.10 \ - opencv-python==4.2.0.32 \ + opencv-python==4.4.0.42 \ git+https://github.com/NVIDIA/dllogger \ yacs @@ -29,6 +29,6 @@ RUN mkdir /results COPY . . RUN pip install -e . -RUN /opt/conda/bin/conda install -y numpy==1.17.4 +RUN /opt/conda/bin/conda install -y numpy ENV OMP_NUM_THREADS=1 diff --git a/PyTorch/Segmentation/MaskRCNN/pytorch/requirements.txt b/PyTorch/Segmentation/MaskRCNN/pytorch/requirements.txt index 872ba7bd..7a9576ef 100644 --- a/PyTorch/Segmentation/MaskRCNN/pytorch/requirements.txt +++ b/PyTorch/Segmentation/MaskRCNN/pytorch/requirements.txt @@ -1,4 +1,4 @@ mlperf-compliance==0.0.10 -opencv-python==4.2.0.32 +opencv-python==4.4.0.42 yacs git+https://github.com/NVIDIA/cocoapi.git@nvidia/master#egg=cocoapi&subdirectory=PythonAPI