|
|
|
|
|
|
|
|
|
|
|
|
|
import gradio as gr |
|
from fastai.vision.all import * |
|
import PIL |
|
import torchvision.transforms as T |
|
|
|
|
|
def get_y(f): |
|
return int(f.parent.name)-1 |
|
|
|
|
|
learn = load_learner('pig-model.pkl') |
|
|
|
def predict_image(img1, img2): |
|
|
|
|
|
img1 = Image.fromarray(img1) |
|
img2 = Image.fromarray(img2) |
|
|
|
|
|
pred1, _, _ = learn.predict(img1) |
|
pred2, _, _ = learn.predict(img2) |
|
|
|
|
|
|
|
if pred1 == pred2: |
|
|
|
return {"是否是同一头猪": "这是同一头猪"} |
|
else: |
|
|
|
return {"是否是同一头猪": "这不是同一头猪"} |
|
|
|
|
|
|
|
|
|
iface = gr.Interface( |
|
fn=predict_image, |
|
inputs=[ |
|
gr.inputs.Image(label="Image 1"), |
|
gr.inputs.Image(label="Image 2") |
|
], |
|
outputs="text", |
|
title="猪照片识别模型", |
|
description="检测两张图片是否是同一只猪!" |
|
) |
|
|
|
|
|
iface.launch() |
|
|
|
|