Skip to content

Commit

Permalink
Add wheel dependency to detectron notebook
Browse files Browse the repository at this point in the history
  • Loading branch information
yatarkan committed May 31, 2024
1 parent effde5a commit e1d3d7f
Showing 1 changed file with 4 additions and 2 deletions.
6 changes: 4 additions & 2 deletions notebooks/detectron2-to-openvino/detectron2-to-openvino.ipynb
Original file line number Diff line number Diff line change
Expand Up @@ -11,7 +11,8 @@
"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",
"\n",
"\n",
"#### Table of contents:\n\n",
"#### Table of contents:\n",
"\n",
"- [Prerequisites](#Prerequisites)\n",
" - [Define helpers for PyTorch model initialization and conversion](#Define-helpers-for-PyTorch-model-initialization-and-conversion)\n",
" - [Prepare input data](#Prepare-input-data)\n",
Expand Down Expand Up @@ -44,6 +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 \"git+https://github.com/facebookresearch/detectron2.git\" --extra-index-url https://download.pytorch.org/whl/cpu\n",
"%pip install -q \"openvino>=2023.1.0\""
Expand Down Expand Up @@ -619,4 +621,4 @@
},
"nbformat": 4,
"nbformat_minor": 4
}
}

0 comments on commit e1d3d7f

Please sign in to comment.