only show 15 prediction images
"source": [
"predictions = model.predict(validation_images)\n",
"\n",
- "for image, mask, prediction in zip(validation_images, validation_masks, predictions):\n",
+ "image_combos = list(zip(validation_images, validation_masks, predictions))\n",
+ "num_to_display = 15\n",
+ "\n",
+ "for image, mask, prediction in image_combos[0:num_to_display]:\n",
" display(image, mask, prediction)\n",
" plt.show()"
]
