Spaces:
Runtime error
Runtime error
File size: 2,951 Bytes
4e87a32 cd38c64 978e03b b08e5f0 978e03b 4e87a32 b08e5f0 978e03b b08e5f0 4e87a32 11dd1e3 b08e5f0 4e87a32 b08e5f0 4e87a32 978e03b 4e87a32 978e03b 4e87a32 11dd1e3 b08e5f0 4e87a32 b08e5f0 4e87a32 978e03b 4e87a32 b08e5f0 4e87a32 b08e5f0 4e87a32 978e03b |
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 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 72 73 74 75 76 77 |
import gradio as gr
from ultralytics import YOLO
model = YOLO('./best.pt') # load your custom trained model
import torch
from ultralyticsplus import render_result
torch.hub.download_url_to_file(
'https://external-content.duckduckgo.com/iu/?u=https%3A%2F%2Ftexashafts.com%2Fwp-content%2Fuploads%2F2016%2F04%2Fconstruction-worker.jpg',
'one.jpg')
torch.hub.download_url_to_file(
'https://www.pearsonkoutcherlaw.com/wp-content/uploads/2020/06/Construction-Workers.jpg', 'two.jpg')
torch.hub.download_url_to_file(
'https://nssgroup.com/wp-content/uploads/2019/02/Building-maintenance-blog.jpg', 'three.jpg')
def yoloV8_func(image: gr.Image = None,
image_size: int = 640,
conf_threshold: float = 0.4,
iou_threshold: float = 0.5):
"""This function performs YOLOv8 object detection on the given image.
Args:
image (gr.Image, optional): Input image to detect objects on. Defaults to None.
image_size (int, optional): Desired image size for the model. Defaults to 640.
conf_threshold (float, optional): Confidence threshold for object detection. Defaults to 0.4.
iou_threshold (float, optional): Intersection over Union threshold for object detection. Defaults to 0.50.
"""
# Load the YOLOv8 model from the 'best.pt' checkpoint
model_path = "yolov8n.pt"
# model = torch.hub.load('ultralytics/yolov8', 'custom', path='/content/best.pt', force_reload=True, trust_repo=True)
# Perform object detection on the input image using the YOLOv8 model
results = model.predict(image,
conf=conf_threshold,
iou=iou_threshold,
imgsz=image_size)
# Print the detected objects' information (class, coordinates, and probability)
box = results[0].boxes
print("Object type:", box.cls)
print("Coordinates:", box.xyxy)
print("Probability:", box.conf)
# Render the output image with bounding boxes around detected objects
render = render_result(model=model, image=image, result=results[0])
return render
inputs = [
gr.Image(type="filepath", label="Input Image"),
gr.Slider(minimum=320, maximum=1280, step=32, label="Image Size", value=640),
gr.Slider(minimum=0.0, maximum=1.0, step=0.05, label="Confidence Threshold"),
gr.Slider(minimum=0.0, maximum=1.0, step=0.05, label="IOU Threshold"),
]
outputs = gr.Image(type="filepath", label="Output Image")
title = "YOLOv8 101: Custom Object Detection on Construction Workers"
examples = [['img1.jpg', 640, 0.5, 0.7],
['img2.jpg', 800, 0.5, 0.6],
['img3.jpg', 900, 0.5, 0.8]]
yolo_app = gr.Interface(
fn=yoloV8_func,
inputs=inputs,
outputs=outputs,
title=title,
examples=examples,
cache_examples=False,
)
# Launch the Gradio interface in debug mode with queue enabled
yolo_app.launch(debug=True, share=True).queue()
|