Allow loading png images in object detection jupyter notebook (tensorflow#5269)

aysark · pkulzc · commit 8ffcc2fa3287 · 2018-09-19T01:22:33.000-07:00
diff --git a/research/object_detection/object_detection_tutorial.ipynb b/research/object_detection/object_detection_tutorial.ipynb
@@ -297,6 +297,8 @@
       "source": [
         "def load_image_into_numpy_array(image):\n",
         "  (im_width, im_height) = image.size\n",
+        "  if image.format == 'PNG':\n",
+        "      image = image.convert('RGB')\n",
         "  return np.array(image.getdata()).reshape(\n",
         "      (im_height, im_width, 3)).astype(np.uint8)"
       ]

Original file line number	Diff line number	Diff line change
`@@ -297,6 +297,8 @@`
`297`	`297`	`"source": [`
`298`	`298`	`"def load_image_into_numpy_array(image):\n",`
`299`	`299`	`" (im_width, im_height) = image.size\n",`
	`300`	`+ " if image.format == 'PNG':\n",`
	`301`	`+ " image = image.convert('RGB')\n",`
`300`	`302`	`" return np.array(image.getdata()).reshape(\n",`
`301`	`303`	`" (im_height, im_width, 3)).astype(np.uint8)"`
`302`	`304`	`]`