File size: 871 Bytes
7eda1e7 65c1b15 a376dac ea07644 65c1b15 a376dac 65c1b15 a376dac 1b92647 a376dac 368ef8a a785cd5 21181e0 0250c84 368ef8a 0250c84 65c1b15 |
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 |
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])
css = """
.gr-example img {
width: 300px !important;
height: auto !important;
}
"""
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']],
examples_per_page=4,
cache_examples= False,
css=css
)
demo.launch()
|