File size: 542 Bytes
d66fb96 9470839 d66fb96 dc05dfb 9470839 |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 |
import torch
import gradio as gr
from PIL import Image
from ultralytics import YOLO
model=YOLO("yolov8s.pt")
def predict_Image(img):
if img.mode!="RGB":
img=img.convert("RGB")
res=model.predict(source=img,conf=0.25)
im_array=res[0].plot()
pil_img=Image.fromarray(im_array[...,::-1])
return pil_img
iface=gr.Interface(fn=predict_Image,inputs=gr.Image(type="pil"),outputs="image",examples=["input_img/detectron.png"],title="Detection of EBIKE on SparseRCNN",description="upload a picture to detect EBIKE")
iface.launch(share=True) |