Commit 9a7c33a3 authored by Vighnesh Birodkar's avatar Vighnesh Birodkar Committed by TF Object Detection Team
Browse files

Restrict DeepMAC colab to install tf-models-official<=2.8.

PiperOrigin-RevId: 466486717
parent 974ae9c4
......@@ -3,7 +3,7 @@
"nbformat_minor": 0,
"metadata": {
"colab": {
"name": "deepmac_demo.ipynb",
"name": "deepmac_colab.ipynb",
"provenance": [],
"collapsed_sections": []
},
......@@ -61,8 +61,6 @@
"id": "nNdls0Pe0UPK"
},
"source": [
"!pip install -U --pre tensorflow==\"2.2.0\"\n",
"\n",
"import os\n",
"import pathlib\n",
"\n",
......@@ -87,7 +85,11 @@
"cd models/research/\n",
"protoc object_detection/protos/*.proto --python_out=.\n",
"cp object_detection/packages/tf2/setup.py .\n",
"python -m pip install ."
"\n",
"# The latest tf-models-official installs tensorflow 2.9 which has an\n",
"# incompatible CuDNN dependency. Here we restrict ourselves to versions 2.8 and\n",
"# below.\n",
"python -m pip install \"tf-models-official<=2.8\" ."
],
"execution_count": null,
"outputs": []
......@@ -326,6 +328,7 @@
"id": "vdzuKnpj1A3L"
},
"source": [
"\n",
"%matplotlib inline\n",
"\n",
"boxes = boxes_list[0]\n",
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment