| | |
| | """app.ipynb |
| | |
| | Automatically generated by Colab. |
| | |
| | Original file is located at |
| | https://colab.research.google.com/drive/1nIR_ak2SXq9fwQmKpq4eupLwQU_SI4fn |
| | """ |
| |
|
| | import gradio as gr |
| | import torch |
| | import cv2 |
| | from ultralytics import YOLO |
| | import numpy as np |
| |
|
| | model = YOLO("best.torchscript") |
| |
|
| | def predict(image): |
| | results = model(image) |
| | annotated = results[0].plot() |
| | annotated_rgb = cv2.cvtColor(annotated, cv2.COLOR_BGR2RGB) |
| | return annotated_rgb |
| |
|
| | iface = gr.Interface( |
| | fn=predict, |
| | inputs=gr.Image(type="numpy", label="Upload an image"), |
| | outputs=gr.Image(type="numpy", label="Detected image"), |
| | title="YOLOv8 Object Detection", |
| | description="Upload an image to detect objects using a YOLOv8 model in TorchScript format.", |
| | examples=None |
| | ) |
| |
|
| | if __name__ == "__main__": |
| | iface.launch() |
