trainer classifer

.ipynb_checkpoints/app-checkpoint.ipynb

@@ -0,0 +1,294 @@
+{
+ "cells": [
+  {
+   "cell_type": "code",
+   "execution_count": 47,
+   "id": "88aaacc6-8379-4b6b-b3b4-eea4d9acaea0",
+   "metadata": {},
+   "outputs": [],
+   "source": [
+    "#|default_exp app"
+   ]
+  },
+  {
+   "cell_type": "markdown",
+   "id": "0d06de33-59f0-4957-87a2-6154102289a6",
+   "metadata": {},
+   "source": [
+    "#### Sneakers"
+   ]
+  },
+  {
+   "cell_type": "code",
+   "execution_count": 65,
+   "id": "844c3071-836e-46c5-b30e-2b8d96a8e369",
+   "metadata": {},
+   "outputs": [],
+   "source": [
+    "#|export\n",
+    "from fastai.vision.all import *\n",
+    "import gradio as gr\n",
+    "from fastai.vision.widgets import *"
+   ]
+  },
+  {
+   "cell_type": "code",
+   "execution_count": 66,
+   "id": "ded0d929-67f6-4a69-b223-dee3ad189139",
+   "metadata": {},
+   "outputs": [],
+   "source": [
+    "# btn_upload = SimpleNamespace(data = ['images/IMG_0115.jpg'])"
+   ]
+  },
+  {
+   "cell_type": "code",
+   "execution_count": 67,
+   "id": "6b7aeaaa-8260-45fa-9178-e064b0132c36",
+   "metadata": {},
+   "outputs": [],
+   "source": [
+    "# img = PILImage.create(btn_upload.data[-1])\n",
+    "# # img.thumbnail((192,192))\n",
+    "# # img"
+   ]
+  },
+  {
+   "cell_type": "code",
+   "execution_count": 68,
+   "id": "a0dfbe70-b826-4903-95cf-ae59c498a3b6",
+   "metadata": {},
+   "outputs": [],
+   "source": [
+    "# out_pl = widgets.Output()\n",
+    "# out_pl.clear_output()\n",
+    "# with out_pl: display(img.to_thumb(128,128))\n",
+    "# out_pl"
+   ]
+  },
+  {
+   "cell_type": "code",
+   "execution_count": 69,
+   "id": "a5aaeeb6-6a93-4de8-be39-36c4b5d90ee4",
+   "metadata": {},
+   "outputs": [],
+   "source": [
+    "# path"
+   ]
+  },
+  {
+   "cell_type": "code",
+   "execution_count": 70,
+   "id": "7fcc53f8-0326-49d0-a110-1140148d6301",
+   "metadata": {},
+   "outputs": [],
+   "source": [
+    "# learn_inf.predict(img)"
+   ]
+  },
+  {
+   "cell_type": "code",
+   "execution_count": 71,
+   "id": "6173ae0f-fa63-4a94-b9f6-e0f9534ff8d9",
+   "metadata": {},
+   "outputs": [],
+   "source": [
+    "# pred,pred_idx,probs = learn_inf.predict(img)"
+   ]
+  },
+  {
+   "cell_type": "code",
+   "execution_count": 72,
+   "id": "490af644-cb3c-4f8c-9e12-1a7cf9c18fc0",
+   "metadata": {},
+   "outputs": [],
+   "source": [
+    "# #hide_output\n",
+    "# lbl_pred = widgets.Label()\n",
+    "# lbl_pred.value = f'Prediction: {pred}; Probability: {probs[pred_idx]:.04f}'\n",
+    "# lbl_pred"
+   ]
+  },
+  {
+   "cell_type": "code",
+   "execution_count": 73,
+   "id": "e3618d64-5db3-4659-b30f-f917939f509a",
+   "metadata": {},
+   "outputs": [],
+   "source": [
+    "# #Putting back btn_upload to a widget for next cell\n",
+    "# btn_upload = widgets.FileUpload()"
+   ]
+  },
+  {
+   "cell_type": "code",
+   "execution_count": 74,
+   "id": "b2475bb3-b820-4a9d-b3b7-488e9e2971bc",
+   "metadata": {},
+   "outputs": [],
+   "source": [
+    "# VBox([widgets.Label('Select your sneakers!'), \n",
+    "#       btn_upload, btn_run, out_pl, lbl_pred])"
+   ]
+  },
+  {
+   "cell_type": "code",
+   "execution_count": 75,
+   "id": "1cdad192-403a-44d4-b006-661afebe18da",
+   "metadata": {},
+   "outputs": [],
+   "source": [
+    "#|export\n",
+    "path = Path('../fastai')\n",
+    "learn_inf = load_learner(path/'export.pkl')"
+   ]
+  },
+  {
+   "cell_type": "code",
+   "execution_count": 76,
+   "id": "c72e52ad-30b5-4c06-83d4-35095e43d419",
+   "metadata": {},
+   "outputs": [
+    {
+     "data": {
+      "application/vnd.jupyter.widget-view+json": {
+       "model_id": "7ec47a7f52a449a398a190e295f76328",
+       "version_major": 2,
+       "version_minor": 0
+      },
+      "text/plain": [
+       "Button(description='Classify', style=ButtonStyle())"
+      ]
+     },
+     "execution_count": 76,
+     "metadata": {},
+     "output_type": "execute_result"
+    }
+   ],
+   "source": [
+    "#hide_output\n",
+    "btn_run = widgets.Button(description='Classify')\n",
+    "btn_run"
+   ]
+  },
+  {
+   "cell_type": "code",
+   "execution_count": 77,
+   "id": "936e0aa5-4344-4b86-9e7d-b21adbd6b5ab",
+   "metadata": {},
+   "outputs": [],
+   "source": [
+    "#|export\n",
+    "def on_click_classify(img):\n",
+    "    # img = PILImage.create(btn_upload.data[-1])\n",
+    "    out_pl = widgets.Output()\n",
+    "    out_pl.clear_output()\n",
+    "    with out_pl: display(img.to_thumb(128,128))\n",
+    "    pred,pred_idx,probs = learn_inf.predict(img)\n",
+    "    # lbl_pred.value = f'Prediction: {pred}; Probability: {probs[pred_idx]:.04f}'\n",
+    "    return f'Prediction: {pred}; Probability: {probs[pred_idx]:.04f}'"
+   ]
+  },
+  {
+   "cell_type": "code",
+   "execution_count": 78,
+   "id": "3bb8c7a0-3fb4-4fbb-9361-c62095a541a8",
+   "metadata": {},
+   "outputs": [],
+   "source": [
+    "# btn_run.on_click(on_click_classify)"
+   ]
+  },
+  {
+   "cell_type": "code",
+   "execution_count": 79,
+   "id": "a6e2a398-f6c8-4444-b0f2-e538a1b36da2",
+   "metadata": {},
+   "outputs": [
+    {
+     "name": "stdout",
+     "output_type": "stream",
+     "text": [
+      "Running on local URL:  http://127.0.0.1:7867\n",
+      "\n",
+      "To create a public link, set `share=True` in `launch()`.\n"
+     ]
+    },
+    {
+     "data": {
+      "text/plain": []
+     },
+     "execution_count": 79,
+     "metadata": {},
+     "output_type": "execute_result"
+    }
+   ],
+   "source": [
+    "#|export\n",
+    "image = gr.Image()\n",
+    "label = gr.Label()\n",
+    "examples = ['images/Adi_trainers.jpg', 'images/Nike_trainers.jpg', 'images/Puma_trainers.jpg', 'images/Adidas_trainers.jpg']\n",
+    "\n",
+    "intf = gr.Interface(fn=on_click_classify, inputs=image, outputs=label, examples=examples)\n",
+    "intf.launch(inline=False)"
+   ]
+  },
+  {
+   "cell_type": "markdown",
+   "id": "16e21449-1821-45a7-829c-acfd92badec1",
+   "metadata": {},
+   "source": [
+    "### Export"
+   ]
+  },
+  {
+   "cell_type": "code",
+   "execution_count": 80,
+   "id": "f5c57105-41fe-4d79-a2b0-d52ed75ea6cb",
+   "metadata": {},
+   "outputs": [
+    {
+     "name": "stdout",
+     "output_type": "stream",
+     "text": [
+      "Export successful\n"
+     ]
+    }
+   ],
+   "source": [
+    "import nbdev\n",
+    "nbdev.export.nb_export('app.ipynb', './')\n",
+    "print('Export successful')"
+   ]
+  },
+  {
+   "cell_type": "code",
+   "execution_count": null,
+   "id": "cb08b3c4-14a5-4e6b-a573-244954914d50",
+   "metadata": {},
+   "outputs": [],
+   "source": []
+  }
+ ],
+ "metadata": {
+  "kernelspec": {
+   "display_name": "Python 3 (ipykernel)",
+   "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.10.12"
+  }
+ },
+ "nbformat": 4,
+ "nbformat_minor": 5
+}

app.ipynb

@@ -0,0 +1,303 @@
+{
+ "cells": [
+  {
+   "cell_type": "code",
+   "execution_count": 1,
+   "id": "88aaacc6-8379-4b6b-b3b4-eea4d9acaea0",
+   "metadata": {},
+   "outputs": [],
+   "source": [
+    "#|default_exp app"
+   ]
+  },
+  {
+   "cell_type": "markdown",
+   "id": "0d06de33-59f0-4957-87a2-6154102289a6",
+   "metadata": {},
+   "source": [
+    "#### Sneakers"
+   ]
+  },
+  {
+   "cell_type": "code",
+   "execution_count": 2,
+   "id": "844c3071-836e-46c5-b30e-2b8d96a8e369",
+   "metadata": {},
+   "outputs": [
+    {
+     "name": "stderr",
+     "output_type": "stream",
+     "text": [
+      "/home/yunix/.local/lib/python3.10/site-packages/matplotlib/projections/__init__.py:63: UserWarning: Unable to import Axes3D. This may be due to multiple versions of Matplotlib being installed (e.g. as a system package and as a pip package). As a result, the 3D projection is not available.\n",
+      "  warnings.warn(\"Unable to import Axes3D. This may be due to multiple versions of \"\n"
+     ]
+    }
+   ],
+   "source": [
+    "#|export\n",
+    "from fastai.vision.all import *\n",
+    "import gradio as gr\n",
+    "from fastai.vision.widgets import *"
+   ]
+  },
+  {
+   "cell_type": "code",
+   "execution_count": 66,
+   "id": "ded0d929-67f6-4a69-b223-dee3ad189139",
+   "metadata": {},
+   "outputs": [],
+   "source": [
+    "# btn_upload = SimpleNamespace(data = ['images/IMG_0115.jpg'])"
+   ]
+  },
+  {
+   "cell_type": "code",
+   "execution_count": 67,
+   "id": "6b7aeaaa-8260-45fa-9178-e064b0132c36",
+   "metadata": {},
+   "outputs": [],
+   "source": [
+    "# img = PILImage.create(btn_upload.data[-1])\n",
+    "# # img.thumbnail((192,192))\n",
+    "# # img"
+   ]
+  },
+  {
+   "cell_type": "code",
+   "execution_count": 68,
+   "id": "a0dfbe70-b826-4903-95cf-ae59c498a3b6",
+   "metadata": {},
+   "outputs": [],
+   "source": [
+    "# out_pl = widgets.Output()\n",
+    "# out_pl.clear_output()\n",
+    "# with out_pl: display(img.to_thumb(128,128))\n",
+    "# out_pl"
+   ]
+  },
+  {
+   "cell_type": "code",
+   "execution_count": 69,
+   "id": "a5aaeeb6-6a93-4de8-be39-36c4b5d90ee4",
+   "metadata": {},
+   "outputs": [],
+   "source": [
+    "# path"
+   ]
+  },
+  {
+   "cell_type": "code",
+   "execution_count": 70,
+   "id": "7fcc53f8-0326-49d0-a110-1140148d6301",
+   "metadata": {},
+   "outputs": [],
+   "source": [
+    "# learn_inf.predict(img)"
+   ]
+  },
+  {
+   "cell_type": "code",
+   "execution_count": 71,
+   "id": "6173ae0f-fa63-4a94-b9f6-e0f9534ff8d9",
+   "metadata": {},
+   "outputs": [],
+   "source": [
+    "# pred,pred_idx,probs = learn_inf.predict(img)"
+   ]
+  },
+  {
+   "cell_type": "code",
+   "execution_count": 72,
+   "id": "490af644-cb3c-4f8c-9e12-1a7cf9c18fc0",
+   "metadata": {},
+   "outputs": [],
+   "source": [
+    "# #hide_output\n",
+    "# lbl_pred = widgets.Label()\n",
+    "# lbl_pred.value = f'Prediction: {pred}; Probability: {probs[pred_idx]:.04f}'\n",
+    "# lbl_pred"
+   ]
+  },
+  {
+   "cell_type": "code",
+   "execution_count": 73,
+   "id": "e3618d64-5db3-4659-b30f-f917939f509a",
+   "metadata": {},
+   "outputs": [],
+   "source": [
+    "# #Putting back btn_upload to a widget for next cell\n",
+    "# btn_upload = widgets.FileUpload()"
+   ]
+  },
+  {
+   "cell_type": "code",
+   "execution_count": 74,
+   "id": "b2475bb3-b820-4a9d-b3b7-488e9e2971bc",
+   "metadata": {},
+   "outputs": [],
+   "source": [
+    "# VBox([widgets.Label('Select your sneakers!'), \n",
+    "#       btn_upload, btn_run, out_pl, lbl_pred])"
+   ]
+  },
+  {
+   "cell_type": "code",
+   "execution_count": 5,
+   "id": "1cdad192-403a-44d4-b006-661afebe18da",
+   "metadata": {},
+   "outputs": [],
+   "source": [
+    "#|export\n",
+    "path = Path('.')\n",
+    "learn_inf = load_learner(path/'export.pkl')"
+   ]
+  },
+  {
+   "cell_type": "code",
+   "execution_count": 6,
+   "id": "c72e52ad-30b5-4c06-83d4-35095e43d419",
+   "metadata": {},
+   "outputs": [
+    {
+     "data": {
+      "application/vnd.jupyter.widget-view+json": {
+       "model_id": "ae40d4f5ffb846c8875f56eeb6c9249c",
+       "version_major": 2,
+       "version_minor": 0
+      },
+      "text/plain": [
+       "Button(description='Classify', style=ButtonStyle())"
+      ]
+     },
+     "execution_count": 6,
+     "metadata": {},
+     "output_type": "execute_result"
+    }
+   ],
+   "source": [
+    "#hide_output\n",
+    "btn_run = widgets.Button(description='Classify')\n",
+    "btn_run"
+   ]
+  },
+  {
+   "cell_type": "code",
+   "execution_count": 7,
+   "id": "936e0aa5-4344-4b86-9e7d-b21adbd6b5ab",
+   "metadata": {},
+   "outputs": [],
+   "source": [
+    "#|export\n",
+    "def on_click_classify(img):\n",
+    "    # img = PILImage.create(btn_upload.data[-1])\n",
+    "    out_pl = widgets.Output()\n",
+    "    out_pl.clear_output()\n",
+    "    with out_pl: display(img.to_thumb(128,128))\n",
+    "    pred,pred_idx,probs = learn_inf.predict(img)\n",
+    "    # lbl_pred.value = f'Prediction: {pred}; Probability: {probs[pred_idx]:.04f}'\n",
+    "    return f'Prediction: {pred}; Probability: {probs[pred_idx]:.04f}'"
+   ]
+  },
+  {
+   "cell_type": "code",
+   "execution_count": 8,
+   "id": "3bb8c7a0-3fb4-4fbb-9361-c62095a541a8",
+   "metadata": {},
+   "outputs": [],
+   "source": [
+    "# btn_run.on_click(on_click_classify)"
+   ]
+  },
+  {
+   "cell_type": "code",
+   "execution_count": 12,
+   "id": "a6e2a398-f6c8-4444-b0f2-e538a1b36da2",
+   "metadata": {},
+   "outputs": [
+    {
+     "name": "stdout",
+     "output_type": "stream",
+     "text": [
+      "Running on local URL:  http://127.0.0.1:7863\n",
+      "\n",
+      "To create a public link, set `share=True` in `launch()`.\n"
+     ]
+    },
+    {
+     "data": {
+      "text/plain": []
+     },
+     "execution_count": 12,
+     "metadata": {},
+     "output_type": "execute_result"
+    }
+   ],
+   "source": [
+    "#|export\n",
+    "image = gr.Image()\n",
+    "label = gr.Label()\n",
+    "examples = ['../images/Adi_trainers.jpg', '../images/Nike_trainers.jpg', '../images/Puma_trainers.jpg', '../images/Adidas_trainers.jpg']\n",
+    "\n",
+    "intf = gr.Interface(fn=on_click_classify, inputs=image, outputs=label, examples=examples)\n",
+    "intf.launch(inline=False)"
+   ]
+  },
+  {
+   "cell_type": "markdown",
+   "id": "16e21449-1821-45a7-829c-acfd92badec1",
+   "metadata": {},
+   "source": [
+    "### Export"
+   ]
+  },
+  {
+   "cell_type": "code",
+   "execution_count": 13,
+   "id": "f5c57105-41fe-4d79-a2b0-d52ed75ea6cb",
+   "metadata": {},
+   "outputs": [
+    {
+     "name": "stdout",
+     "output_type": "stream",
+     "text": [
+      "Export successful\n"
+     ]
+    }
+   ],
+   "source": [
+    "from nbdev.export import nb_export\n",
+    "nb_export('app.ipynb', './')\n",
+    "print('Export successful')"
+   ]
+  },
+  {
+   "cell_type": "code",
+   "execution_count": null,
+   "id": "cb08b3c4-14a5-4e6b-a573-244954914d50",
+   "metadata": {},
+   "outputs": [],
+   "source": []
+  }
+ ],
+ "metadata": {
+  "kernelspec": {
+   "display_name": "Python 3 (ipykernel)",
+   "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.10.12"
+  }
+ },
+ "nbformat": 4,
+ "nbformat_minor": 5
+}

app.py

@@ -1,7 +1,31 @@
+# AUTOGENERATED! DO NOT EDIT! File to edit: app.ipynb.
+
+# %% auto 0
+__all__ = ['path', 'learn_inf', 'image', 'label', 'examples', 'intf', 'on_click_classify']
+
+# %% app.ipynb 2
+from fastai.vision.all import *
 import gradio as gr
+from fastai.vision.widgets import *
+
+# %% app.ipynb 12
+path = Path('.')
+learn_inf = load_learner(path/'export.pkl')
+
+# %% app.ipynb 14
+def on_click_classify(img):
+    # img = PILImage.create(btn_upload.data[-1])
+    out_pl = widgets.Output()
+    out_pl.clear_output()
+    with out_pl: display(img.to_thumb(128,128))
+    pred,pred_idx,probs = learn_inf.predict(img)
+    # lbl_pred.value = f'Prediction: {pred}; Probability: {probs[pred_idx]:.04f}'
+    return f'Prediction: {pred}; Probability: {probs[pred_idx]:.04f}'
 
-def greet(name):
-    return "Hello " + name + "!!"
+# %% app.ipynb 16
+image = gr.Image()
+label = gr.Label()
+examples = ['../images/Adi_trainers.jpg', '../images/Nike_trainers.jpg', '../images/Puma_trainers.jpg', '../images/Adidas_trainers.jpg']
 
-demo = gr.Interface(fn=greet, inputs="text", outputs="text")
-demo.launch()
+intf = gr.Interface(fn=on_click_classify, inputs=image, outputs=label, examples=examples)
+intf.launch(inline=False)

export.pkl

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:3d0301f60385c5509173862e4eb83b572112931ac8a859726b936ab2a921d1ff
+size 46970622