SceneUnderstanding/ADE20K
1
0
Fork 0
Code Issues Pull Requests Projects Releases Wiki Activity

minor comment

Browse Source
This commit is contained in:
xavierpuigf 2021-03-21 17:09:59 -04:00
parent 06f0da1447
commit 0aaae2cb53
1 changed files with 5 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

6
notebooks/ade20k_starter.ipynb
View File

@ -86,7 +86,9 @@
{ {
"cell_type": "code", "cell_type": "code",
"execution_count": 6, "execution_count": 6,
"metadata": {}, "metadata": {
"scrolled": true
},
"outputs": [ "outputs": [
{ {
"name": "stdout", "name": "stdout",
@ -189,6 +191,8 @@
"img = cv2.imread(info['img_name'])[:,:,::-1]\n", "img = cv2.imread(info['img_name'])[:,:,::-1]\n",
"seg = cv2.imread(info['segm_name'])[:,:,::-1]\n", "seg = cv2.imread(info['segm_name'])[:,:,::-1]\n",
"seg_mask = seg.copy()\n", "seg_mask = seg.copy()\n",
"\n",
"# The 0 index in seg_mask corresponds to background (not annotated) pixels\n",
"seg_mask[info['class_mask'] != obj_id+1] *= 0\n", "seg_mask[info['class_mask'] != obj_id+1] *= 0\n",
"plt.figure(figsize=(15,5))\n", "plt.figure(figsize=(15,5))\n",
"\n", "\n",