Spaces:
Running
Running
Update app.py
Browse files
app.py
CHANGED
@@ -1,7 +1,3 @@
|
|
1 |
-
!pip install -U torch torchvision cython
|
2 |
-
!pip install -U 'git+https://github.com/cocodataset/cocoapi.git#subdirectory=PythonAPI'
|
3 |
-
!pip install gradio
|
4 |
-
|
5 |
import torch
|
6 |
import torchvision
|
7 |
from torchvision.transforms import functional as F
|
@@ -9,19 +5,13 @@ from PIL import Image
|
|
9 |
import cv2
|
10 |
import gradio as gr
|
11 |
|
12 |
-
!git clone https://github.com/ultralytics/yolov5
|
13 |
-
%cd yolov5
|
14 |
-
!pip install -r requirements.txt
|
15 |
-
|
16 |
-
import torch
|
17 |
from yolov5.models.yolo import Model
|
|
|
18 |
|
19 |
device = torch.device('cuda' if torch.cuda.is_available() else 'cpu')
|
20 |
model = torch.hub.load('ultralytics/yolov5', 'yolov5s', pretrained=True).to(device)
|
21 |
model.eval()
|
22 |
|
23 |
-
from yolov5.utils.general import non_max_suppression
|
24 |
-
|
25 |
def preprocess_image(image):
|
26 |
image_tensor = F.to_tensor(image)
|
27 |
return image_tensor.unsqueeze(0).to(device)
|
|
|
|
|
|
|
|
|
|
|
1 |
import torch
|
2 |
import torchvision
|
3 |
from torchvision.transforms import functional as F
|
|
|
5 |
import cv2
|
6 |
import gradio as gr
|
7 |
|
|
|
|
|
|
|
|
|
|
|
8 |
from yolov5.models.yolo import Model
|
9 |
+
from yolov5.utils.general import non_max_suppression
|
10 |
|
11 |
device = torch.device('cuda' if torch.cuda.is_available() else 'cpu')
|
12 |
model = torch.hub.load('ultralytics/yolov5', 'yolov5s', pretrained=True).to(device)
|
13 |
model.eval()
|
14 |
|
|
|
|
|
15 |
def preprocess_image(image):
|
16 |
image_tensor = F.to_tensor(image)
|
17 |
return image_tensor.unsqueeze(0).to(device)
|