The Super Duper Bear Classification Algorithm¶

Here I am implementing the bear detector model as a binder notebook

In [2]:

# What you need
from fastai.vision.all import *
from fastai.vision.widgets import *

In [3]:

path = Path()
learn_inf = load_learner(path/'export.pkl', cpu=True)
btn_upload = widgets.FileUpload()
out_pl = widgets.Output()
lbl_pred = widgets.Label()

In [4]:

def on_data_change(change):
    lbl_pred.value = ''
    img = PILImage.create(btn_upload.data[-1])
    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}'

In [5]:

btn_upload.observe(on_data_change, names=['data'])

In [6]:

display(VBox([widgets.Label('Select your bear!'), btn_upload, out_pl, lbl_pred]))

VBox(children=(Label(value='Select your bear!'), FileUpload(value={}, description='Upload'), Output(), Label(v…

In [ ]: