galatasaray commited on
Commit
2358c71
1 Parent(s): ed3a4b5

file path error

Browse files
Files changed (1) hide show
  1. app.py +8 -5
app.py CHANGED
@@ -11,9 +11,11 @@ cwd = os.getcwd()
11
 
12
 
13
  # Images
14
- #torch.hub.download_url_to_file('https://github.com/ultralytics/yolov5/raw/master/data/images/zidane.jpg', 'zidane.jpg')
15
- #torch.hub.download_url_to_file('https://github.com/ultralytics/yolov5/raw/master/data/images/bus.jpg', 'bus.jpg')
16
-
 
 
17
  # Model
18
  #model = torch.hub.load('ultralytics/yolov5', 'yolov5s') # force_reload=True to update
19
  model = torch.hub.load(cwd+'/yolov5', 'custom', path=cwd+'/saved_model/s1000_best.pt', source='local') # local model
@@ -38,8 +40,9 @@ article = "<p style='text-align: center'>YOLOv5 is a family of compound-scaled o
38
  "and export to ONNX, CoreML and TFLite. <a href='https://github.com/ultralytics/yolov5'>Source code</a> |" \
39
  "<a href='https://apps.apple.com/app/id1452689527'>iOS App</a> | <a href='https://pytorch.org/hub/ultralytics_yolov5'>PyTorch Hub</a></p>"
40
 
41
- path_folder = cwd+'/datasets/s1000/'
42
- examples = [[path_folder+'s1000_50_metre.jpg'], [path_folder+'s1000_100_metre.jpg'],[path_folder+'s1000_150_metre.jpg'],[path_folder+'s1000_200_metre.jpg'],[path_folder+'s1000_250_metre.jpg']]
 
43
  gr.Interface(yolo, inputs, outputs, title=title, description=description, article=article, examples=examples, analytics_enabled=False).launch(
44
  debug=True)
45
 
 
11
 
12
 
13
  # Images
14
+ torch.hub.download_url_to_file('https://i.imgur.com/Qi6I5Yf.png', 's1000_50_metre.jpg')
15
+ torch.hub.download_url_to_file('https://i.imgur.com/kATwqiq.png', 's1000_100_metre.jpg')
16
+ torch.hub.download_url_to_file('https://i.imgur.com/UhtsKNx.png', 's1000_150_metre.jpg')
17
+ torch.hub.download_url_to_file('https://i.imgur.com/xXnXoEX.png', 's1000_200_metre.jpg')
18
+ torch.hub.download_url_to_file('https://i.imgur.com/SU5yPAS.png', 's1000_250_metre.jpg')
19
  # Model
20
  #model = torch.hub.load('ultralytics/yolov5', 'yolov5s') # force_reload=True to update
21
  model = torch.hub.load(cwd+'/yolov5', 'custom', path=cwd+'/saved_model/s1000_best.pt', source='local') # local model
 
40
  "and export to ONNX, CoreML and TFLite. <a href='https://github.com/ultralytics/yolov5'>Source code</a> |" \
41
  "<a href='https://apps.apple.com/app/id1452689527'>iOS App</a> | <a href='https://pytorch.org/hub/ultralytics_yolov5'>PyTorch Hub</a></p>"
42
 
43
+ '''path_folder = cwd+'/datasets/s1000/'
44
+ examples = [[path_folder+'s1000_50_metre.jpg'], [path_folder+'s1000_100_metre.jpg'],[path_folder+'s1000_150_metre.jpg'],[path_folder+'s1000_200_metre.jpg'],[path_folder+'s1000_250_metre.jpg']]'''
45
+ examples = [['s1000_50_metre.jpg'], ['s1000_100_metre.jpg'],['s1000_150_metre.jpg'],['s1000_200_metre.jpg'],['s1000_250_metre.jpg']]
46
  gr.Interface(yolo, inputs, outputs, title=title, description=description, article=article, examples=examples, analytics_enabled=False).launch(
47
  debug=True)
48