chris/fastbook
1
0
Fork 0
mirror of https://github.com/fastai/fastbook.git synced 2025-04-06 10:50:44 +00:00
Code Issues Packages Projects Releases Wiki Activity

learn.predict throws error when passed PIL image

Browse Source
This commit is contained in:
yousif 2023-03-05 12:37:47 -05:00
parent 823b69e00a
commit 92b59e14e9
1 changed files with 1 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

3
01_intro.ipynb
View File

@ -758,8 +758,7 @@
} }
], ],
"source": [ "source": [
"img = PILImage.create(uploader.data[0])\n", "is_cat,_,probs = learn.predict(uploader.data[0])\n",
"is_cat,_,probs = learn.predict(img)\n",
"print(f\"Is this a cat?: {is_cat}.\")\n", "print(f\"Is this a cat?: {is_cat}.\")\n",
"print(f\"Probability it's a cat: {probs[1].item():.6f}\")" "print(f\"Probability it's a cat: {probs[1].item():.6f}\")"
] ]