File size: 642 Bytes
7eda1e7 65c1b15 a376dac ea07644 65c1b15 a376dac 65c1b15 a376dac 1b92647 a376dac a785cd5 21181e0 65c1b15 |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 |
import cv2
import gradio as gr
from ultralytics import YOLO
from PIL import Image
model = YOLO('hotspot_detector.pt')
def detect_hotspots(image):
result = model(image)
for r in result:
im_array = r.plot()
# im = Image.fromarray(im_array[..., ::-1])
return Image.fromarray(im_array[..., ::-1])
demo = gr.Interface(fn=detect_hotspots, inputs=gr.Image(type='pil'), outputs="image", title="YOLO Object Detection",
examples=[['images/test_image_1.jpg'], ['images/test_image_2.jpg'],
['images/test_image_3.jpg'], ['images/test_image_4.jpg']])
demo.launch()
|