diff --git "a/demo.ipynb" "b/demo.ipynb" new file mode 100644--- /dev/null +++ "b/demo.ipynb" @@ -0,0 +1,103 @@ +{ + "cells": [ + { + "cell_type": "code", + "execution_count": 1, + "id": "2f6c7352", + "metadata": {}, + "outputs": [ + { + "data": { + "application/vnd.jupyter.widget-view+json": { + "model_id": "814fabf688404b1a89fee4dec46a7f77", + "version_major": 2, + "version_minor": 0 + }, + "text/plain": [ + "Fetching 1 files: 0%| | 0/1 [00:00" + ] + }, + "metadata": {}, + "output_type": "display_data" + } + ], + "source": [ + "fig, axes = plt.subplots(1, len(images), figsize=(12, 5))\n", + "\n", + "for i, (ax, img) in enumerate(zip(axes, images)):\n", + " ax.imshow(img, cmap='gray')\n", + " ax.axis('off')\n", + " ax.set_title(f\"Class: {predicted_classes[i]}\\nConfidence: {confidences[i]:.3f}\", \n", + " fontsize=12, fontweight='bold')\n", + "\n", + "plt.tight_layout()\n", + "plt.show()" + ] + }, + { + "cell_type": "code", + "execution_count": null, + "id": "3d61b960", + "metadata": {}, + "outputs": [], + "source": [] + } + ], + "metadata": { + "kernelspec": { + "display_name": "dino_plankton", + "language": "python", + "name": "python3" + }, + "language_info": { + "codemirror_mode": { + "name": "ipython", + "version": 3 + }, + "file_extension": ".py", + "mimetype": "text/x-python", + "name": "python", + "nbconvert_exporter": "python", + "pygments_lexer": "ipython3", + "version": "3.12.11" + } + }, + "nbformat": 4, + "nbformat_minor": 5 +}