diff --git a/Dockerfile b/Dockerfile index d5777104c8..45673fd775 100644 --- a/Dockerfile +++ b/Dockerfile @@ -22,6 +22,9 @@ RUN git clone --recursive https://github.com/zarr-developers/numcodecs.git && pi WORKDIR /opt/monai +# remove opencv-python before opencv-python-headless installation +RUN pip uninstall -y opencv && rm /usr/local/lib/python3.10/dist-packages/cv2 -r + # install full deps COPY requirements.txt requirements-min.txt requirements-dev.txt /tmp/ RUN cp /tmp/requirements.txt /tmp/req.bak \ diff --git a/monai/data/video_dataset.py b/monai/data/video_dataset.py index be3bcf5bd5..ed5c37d777 100644 --- a/monai/data/video_dataset.py +++ b/monai/data/video_dataset.py @@ -177,7 +177,7 @@ def get_available_codecs() -> dict[str, str]: with tempfile.TemporaryDirectory() as tmp_dir: for codec, ext in all_codecs.items(): fname = os.path.join(tmp_dir, f"test{ext}") - fourcc = cv2.VideoWriter_fourcc(*codec) + fourcc = cv2.VideoWriter_fourcc(*codec) # type: ignore[attr-defined] noviderr = writer.open(fname, fourcc, 1, (10, 10)) if noviderr: codecs[codec] = ext diff --git a/requirements-dev.txt b/requirements-dev.txt index af1b8b89d5..2855b56ad5 100644 --- a/requirements-dev.txt +++ b/requirements-dev.txt @@ -57,3 +57,4 @@ zarr lpips==0.1.4 nvidia-ml-py huggingface_hub +opencv-python-headless