Skip to content

Commit e1d3d7f

Browse files
committed
Add wheel dependency to detectron notebook
1 parent effde5a commit e1d3d7f

File tree

1 file changed

+4
-2
lines changed

1 file changed

+4
-2
lines changed

notebooks/detectron2-to-openvino/detectron2-to-openvino.ipynb

Lines changed: 4 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -11,7 +11,8 @@
1111
"In this tutorial we consider how to convert and run Detectron2 models using OpenVINO™. We will use `Faster R-CNN FPN x1` model and `Mask R-CNN FPN x3` pretrained on [COCO](https://cocodataset.org/#home) dataset as examples for object detection and instance segmentation respectively.\n",
1212
"\n",
1313
"\n",
14-
"#### Table of contents:\n\n",
14+
"#### Table of contents:\n",
15+
"\n",
1516
"- [Prerequisites](#Prerequisites)\n",
1617
" - [Define helpers for PyTorch model initialization and conversion](#Define-helpers-for-PyTorch-model-initialization-and-conversion)\n",
1718
" - [Prepare input data](#Prepare-input-data)\n",
@@ -44,6 +45,7 @@
4445
"metadata": {},
4546
"outputs": [],
4647
"source": [
48+
"%pip install -q wheel\n",
4749
"%pip install -q --extra-index-url https://download.pytorch.org/whl/cpu torch torchvision\n",
4850
"%pip install -q \"git+https://github.com/facebookresearch/detectron2.git\" --extra-index-url https://download.pytorch.org/whl/cpu\n",
4951
"%pip install -q \"openvino>=2023.1.0\""
@@ -619,4 +621,4 @@
619621
},
620622
"nbformat": 4,
621623
"nbformat_minor": 4
622-
}
624+
}

0 commit comments

Comments
 (0)