Spaces:
Runtime error
Runtime error
Delete app.py
#2
by
Ultrajonic
- opened
app.py
DELETED
|
@@ -1,31 +0,0 @@
|
|
| 1 |
-
import gradio as gr
|
| 2 |
-
import torch
|
| 3 |
-
from PIL import Image
|
| 4 |
-
import os
|
| 5 |
-
# Images
|
| 6 |
-
torch.hub.download_url_to_file('https://github.com/ultralytics/yolov5/raw/master/data/images/zidane.jpg', 'zidane.jpg')
|
| 7 |
-
torch.hub.download_url_to_file('https://github.com/ultralytics/yolov5/raw/master/data/images/bus.jpg', 'bus.jpg')
|
| 8 |
-
|
| 9 |
-
# Model
|
| 10 |
-
model = torch.hub.load('ultralytics/yolov5', 'yolov5s') # force_reload=True to update
|
| 11 |
-
|
| 12 |
-
def yolo(im, size=640):
|
| 13 |
-
g = (size / max(im.size)) # gain
|
| 14 |
-
im = im.resize((int(x * g) for x in im.size), Image.ANTIALIAS) # resize
|
| 15 |
-
|
| 16 |
-
results = model(im) # inference
|
| 17 |
-
results.render() # updates results.imgs with boxes and labels
|
| 18 |
-
results.save()
|
| 19 |
-
os.system("ls")
|
| 20 |
-
return "out.png"
|
| 21 |
-
|
| 22 |
-
|
| 23 |
-
inputs = gr.inputs.Image(type='pil', label="Original Image")
|
| 24 |
-
outputs = gr.outputs.Image(type="file", label="Output Image")
|
| 25 |
-
|
| 26 |
-
title = "YOLOv5"
|
| 27 |
-
description = "YOLOv5 Gradio demo for object detection. Upload an image or click an example image to use."
|
| 28 |
-
article = "<p style='text-align: center'>YOLOv5 is a family of compound-scaled object detection models trained on the COCO dataset, and includes simple functionality for Test Time Augmentation (TTA), model ensembling, hyperparameter evolution, and export to ONNX, CoreML and TFLite. <a href='https://github.com/ultralytics/yolov5'>Source code</a> | <a href='https://pytorch.org/hub/ultralytics_yolov5'>PyTorch Hub</a></p>"
|
| 29 |
-
|
| 30 |
-
examples = [['zidane.jpg'], ['bus.jpg']]
|
| 31 |
-
gr.Interface(yolo, inputs, outputs, title=title, description=description, article=article, examples=examples, theme="huggingface").launch(cache_examples=True,enable_queue=True)
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|