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
......@@ -30,7 +30,9 @@
{
"cell_type": "code",
"execution_count": null,
"metadata": {},
"metadata": {
"collapsed": true
},
"outputs": [],
"source": [
"import collections\n",
......@@ -69,7 +71,9 @@
{
"cell_type": "code",
"execution_count": null,
"metadata": {},
"metadata": {
"collapsed": true
},
"outputs": [],
"source": [
"_MODEL_URLS = {\n",
......@@ -89,7 +93,9 @@
{
"cell_type": "code",
"execution_count": null,
"metadata": {},
"metadata": {
"collapsed": true
},
"outputs": [],
"source": [
"# Check configuration and download the model\n",
......@@ -117,7 +123,9 @@
{
"cell_type": "code",
"execution_count": null,
"metadata": {},
"metadata": {
"collapsed": true
},
"outputs": [],
"source": [
"_FROZEN_GRAPH_NAME = 'frozen_inference_graph'\n",
......@@ -177,7 +185,9 @@
{
"cell_type": "code",
"execution_count": null,
"metadata": {},
"metadata": {
"collapsed": true
},
"outputs": [],
"source": [
"model = DeepLabModel(download_path)"
......@@ -193,7 +203,9 @@
{
"cell_type": "code",
"execution_count": null,
"metadata": {},
"metadata": {
"collapsed": true
},
"outputs": [],
"source": [
"LABEL_NAMES = np.asarray([\n",
......@@ -235,7 +247,8 @@
" ax.yaxis.tick_right()\n",
" plt.yticks(range(len(unique_labels)), LABEL_NAMES[unique_labels])\n",
" plt.xticks([], [])\n",
" ax.tick_params(width=0)\n"
" ax.tick_params(width=0)\n",
" plt.show()"
]
},
{
......@@ -248,7 +261,9 @@
{
"cell_type": "code",
"execution_count": null,
"metadata": {},
"metadata": {
"collapsed": true
},
"outputs": [],
"source": [
"# Note that we are using single scale inference in the demo for fast\n",
......@@ -283,6 +298,7 @@
"cell_type": "code",
"execution_count": null,
"metadata": {
"collapsed": true,
"scrolled": true
},
"outputs": [],
......@@ -313,7 +329,9 @@
{
"cell_type": "code",
"execution_count": null,
"metadata": {},
"metadata": {
"collapsed": true
},
"outputs": [],
"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