File size: 947 Bytes
8141dc9 29fa556 8141dc9 |
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 29 30 31 32 33 34 35 36 37 38 39 |
from ultralytics import YOLO
import cv2
import gradio as gr
file = "my_model.pt"
images = [["image1.jpg"],["image2.jpg"]]
model = YOLO(file)
def show_preds_image(image_path):
image = cv2. imread(image_path)
outputs = model.predict(source=image_path)
results = outputs[0]. cpu().numpy()
for i, det in enumerate(results.boxes.xyxy):
cv2. rectangle(image,(int(det[0]), int(det[1])), (int(det[2]), int(det[3])),
thickness=6 , color=[0,0,255]
)
return cv2. cvtColor (image, cv2.COLOR_BGR2RGB)
inputs_image = [
gr.components.Image(type="filepath", label="Input Image"),
]
outputs_image = [
gr.components.Image(type="numpy", label="Output Image"),
]
interface_image = gr.Interface(
fn=show_preds_image,
inputs=inputs_image,
outputs=outputs_image,
title="Plastic Detector",
examples=images,
cache_examples=False,
)
interface_image.launch()
|