| 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) |