|
import torch |
|
from PIL import Image |
|
import gradio as gr |
|
|
|
|
|
model = torch.hub.load('ultralytics/yolov5', 'custom', path='best.pt', force_reload=True) |
|
model.conf = 0.6 |
|
model.iou = 0.2 |
|
|
|
|
|
def predict(image): |
|
|
|
results = model(image, size=1088) |
|
|
|
|
|
annotated_image = results.render() |
|
|
|
|
|
|
|
labeled_image = Image.fromarray(annotated_image[0]) if annotated_image else Image.fromarray(np.array(image)) |
|
num_objects = len(results.xyxy[0]) |
|
|
|
return labeled_image, num_objects |
|
|
|
|
|
iface = gr.Interface( |
|
fn=predict, |
|
inputs=gr.components.Image(type='pil', label="上传mwr~~的图片"), |
|
outputs=[gr.components.Image(type='pil', label="含标记的图片结果"), gr.components.Label(label="总币数")], |
|
title="数金币", |
|
description="只能识别个数" |
|
) |
|
|
|
|
|
iface.launch(debug=True) |
|
|