Add opencv dependency to detectron notebook

openvinotoolkit · May 31, 2024 · 654eb32 · 654eb32
1 parent e1d3d7f
commit 654eb32
Showing 1 changed file with 1 addition and 2 deletions.
diff --git a/notebooks/detectron2-to-openvino/detectron2-to-openvino.ipynb b/notebooks/detectron2-to-openvino/detectron2-to-openvino.ipynb
@@ -45,8 +45,7 @@
    "metadata": {},
    "outputs": [],
    "source": [
-    "%pip install -q wheel\n",
-    "%pip install -q --extra-index-url https://download.pytorch.org/whl/cpu torch torchvision\n",
+    "%pip install -q \"torch\" \"torchvision\" \"opencv-python\" \"wheel\" --extra-index-url https://download.pytorch.org/whl/cpu \n",
     "%pip install -q \"git+https://github.com/facebookresearch/detectron2.git\" --extra-index-url https://download.pytorch.org/whl/cpu\n",
     "%pip install -q \"openvino>=2023.1.0\""
    ]