akhaliq HF staff commited on
Commit
f531a83
1 Parent(s): 2611d91

Create app.py

Browse files
Files changed (1) hide show
  1. app.py +33 -0
app.py ADDED
@@ -0,0 +1,33 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ from PIL import Image
2
+ import torch
3
+ from torchvision import transforms
4
+
5
+ # load model
6
+ model = torch.hub.load('hustvl/yolop', 'yolop', pretrained=True)
7
+
8
+
9
+ normalize = transforms.Normalize(
10
+ mean=[0.485, 0.456, 0.406], std=[0.229, 0.224, 0.225]
11
+ )
12
+
13
+ transform=transforms.Compose([
14
+ transforms.ToTensor(),
15
+ # normalize
16
+ ])
17
+
18
+
19
+ def inference(img):
20
+
21
+ # print(img.size)
22
+ img = img.resize((640, 640))
23
+
24
+ img = torch.unsqueeze(transform(img), dim=0)
25
+
26
+ # img = transform(img)
27
+
28
+ det_out, da_seg_out,ll_seg_out = model(img)
29
+ ll_out = ll_seg_out[0][0, :, :].detach().numpy()
30
+ da_out = da_seg_out[0][0, :, :].detach().numpy()
31
+ return da_out,ll_out
32
+
33
+ gr.Interface(inference,gr.inputs.Image(type="pil"),["image","image"]).launch(debug=True)