Spaces:
Runtime error
Runtime error
Commit
·
701d992
1
Parent(s):
efe0532
Delete app.py
Browse files
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)
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|