bird_classifier.py

# ---
# jupyter:
#   jupytext:
#     cell_metadata_filter: -all
#     custom_cell_magics: kql
#     text_representation:
#       extension: .py
#       format_name: percent
#       format_version: '1.3'
#       jupytext_version: 1.11.2
#   kernelspec:
#     display_name: Python 3.9.14 64-bit
#     language: python
#     name: python3
# ---

# %%
from fastai.vision.all import *

# %%
learn = load_learner("model.pkl")

# %%
categories = ("Bird", "Forest")

# %%
def classify_image(img):
    pred,idx,probs = learn.predict(img)
    print(probs)
    print(*map(float,probs))
    print(*zip(categories, map(float,probs)))
    print(dict(zip(categories, map(float,probs))))
    return dict(zip(categories, map(float,probs)))

# %% [markdown]
#
# im = PILImage.create("bird.jpg")
# im.to_thumb(256, 256)
# im

# %% [markdown]
# classify_image(im)