Skip to content

GitLab

Unverified Commit d3067c21 authored by Yukun Zhu's avatar Yukun Zhu Committed by GitHub
Browse files

Merge pull request #3620 from YogeshGK/master 

plot shown in vis_segmentation
parents d8ed4cba db5fcfad
Hide whitespace changes
Inline Side-by-side
Showing with 27 additions and 9 deletions
research/deeplab/deeplab_demo.ipynb
View file @ d3067c21
...@@ -30,7 +30,9 @@ ...@@ -30,7 +30,9 @@
{ {
"cell_type": "code", "cell_type": "code",
"execution_count": null, "execution_count": null,
"metadata": {}, "metadata": {
"collapsed": true
},
"outputs": [], "outputs": [],
"source": [ "source": [
"import collections\n", "import collections\n",
...@@ -69,7 +71,9 @@ ...@@ -69,7 +71,9 @@
{ {
"cell_type": "code", "cell_type": "code",
"execution_count": null, "execution_count": null,
"metadata": {}, "metadata": {
"collapsed": true
},
"outputs": [], "outputs": [],
"source": [ "source": [
"_MODEL_URLS = {\n", "_MODEL_URLS = {\n",
...@@ -89,7 +93,9 @@ ...@@ -89,7 +93,9 @@
{ {
"cell_type": "code", "cell_type": "code",
"execution_count": null, "execution_count": null,
"metadata": {}, "metadata": {
"collapsed": true
},
"outputs": [], "outputs": [],
"source": [ "source": [
"# Check configuration and download the model\n", "# Check configuration and download the model\n",
...@@ -117,7 +123,9 @@ ...@@ -117,7 +123,9 @@
{ {
"cell_type": "code", "cell_type": "code",
"execution_count": null, "execution_count": null,
"metadata": {}, "metadata": {
"collapsed": true
},
"outputs": [], "outputs": [],
"source": [ "source": [
"_FROZEN_GRAPH_NAME = 'frozen_inference_graph'\n", "_FROZEN_GRAPH_NAME = 'frozen_inference_graph'\n",
...@@ -177,7 +185,9 @@ ...@@ -177,7 +185,9 @@
{ {
"cell_type": "code", "cell_type": "code",
"execution_count": null, "execution_count": null,
"metadata": {}, "metadata": {
"collapsed": true
},
"outputs": [], "outputs": [],
"source": [ "source": [
"model = DeepLabModel(download_path)" "model = DeepLabModel(download_path)"
...@@ -193,7 +203,9 @@ ...@@ -193,7 +203,9 @@
{ {
"cell_type": "code", "cell_type": "code",
"execution_count": null, "execution_count": null,
"metadata": {}, "metadata": {
"collapsed": true
},
"outputs": [], "outputs": [],
"source": [ "source": [
"LABEL_NAMES = np.asarray([\n", "LABEL_NAMES = np.asarray([\n",
...@@ -235,7 +247,8 @@ ...@@ -235,7 +247,8 @@
" ax.yaxis.tick_right()\n", " ax.yaxis.tick_right()\n",
" plt.yticks(range(len(unique_labels)), LABEL_NAMES[unique_labels])\n", " plt.yticks(range(len(unique_labels)), LABEL_NAMES[unique_labels])\n",
" plt.xticks([], [])\n", " plt.xticks([], [])\n",
" ax.tick_params(width=0)\n" " ax.tick_params(width=0)\n",
" plt.show()"
] ]
}, },
{ {
...@@ -248,7 +261,9 @@ ...@@ -248,7 +261,9 @@
{ {
"cell_type": "code", "cell_type": "code",
"execution_count": null, "execution_count": null,
"metadata": {}, "metadata": {
"collapsed": true
},
"outputs": [], "outputs": [],
"source": [ "source": [
"# Note that we are using single scale inference in the demo for fast\n", "# Note that we are using single scale inference in the demo for fast\n",
...@@ -283,6 +298,7 @@ ...@@ -283,6 +298,7 @@
"cell_type": "code", "cell_type": "code",
"execution_count": null, "execution_count": null,
"metadata": { "metadata": {
"collapsed": true,
"scrolled": true "scrolled": true
}, },
"outputs": [], "outputs": [],
...@@ -313,7 +329,9 @@ ...@@ -313,7 +329,9 @@
{ {
"cell_type": "code", "cell_type": "code",
"execution_count": null, "execution_count": null,
"metadata": {}, "metadata": {
"collapsed": true
},
"outputs": [], "outputs": [],
"source": [] "source": []
} }
......
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