minor comment
This commit is contained in:
parent
06f0da1447
commit
0aaae2cb53
|
@ -86,7 +86,9 @@
|
|||
{
|
||||
"cell_type": "code",
|
||||
"execution_count": 6,
|
||||
"metadata": {},
|
||||
"metadata": {
|
||||
"scrolled": true
|
||||
},
|
||||
"outputs": [
|
||||
{
|
||||
"name": "stdout",
|
||||
|
@ -189,6 +191,8 @@
|
|||
"img = cv2.imread(info['img_name'])[:,:,::-1]\n",
|
||||
"seg = cv2.imread(info['segm_name'])[:,:,::-1]\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",
|
||||
"plt.figure(figsize=(15,5))\n",
|
||||
"\n",
|
||||
|
|
Loading…
Reference in New Issue
Block a user