File size: 956 Bytes
5db7e60
 
b7b3dd2
5db7e60
 
 
a4a3b4f
 
 
 
5db7e60
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
import gradio as gr
import os
pkl = "all_20210912_res34.pkl"
from fastai.vision.all import *
from fastai.vision.widgets import *
import jaconv
import pathlib
plt = platform.system()
if plt == 'Linux': pathlib.WindowsPath = pathlib.PosixPath
model_inf = load_learner(pkl)
print(os.getcwd())
def kinoko_uranai(img):
    replace_dic = {"_ッロウッ":" (group)","ー":""}
    result_dic = {}
    pred_class, pred_idxs, outputs = model_inf.predict(img)
    top_5_conf, i = outputs.topk(5)
    itr = 0
    classes = model_inf.dls.vocab
    result_dic = {}
    for x in i:
        kwamei = jaconv.alphabet2kata(classes[x.item()].lower())
        for k,v in replace_dic.items():
            kwamei = kwamei.replace(k,v)
        result_dic[kwamei] = str(round(top_5_conf[itr].item(),2))
        itr=itr+1
    return result_dic
outputs = gr.outputs.Label(num_top_classes=5)
iface = gr.Interface(fn=kinoko_uranai, inputs="image", outputs=outputs).launch(debug=True)