Mauricio Guerta commited on
Commit
3fd22b0
1 Parent(s): 4364716
Files changed (2) hide show
  1. app.py +1 -1
  2. app1.py +1 -1
app.py CHANGED
@@ -8,7 +8,7 @@ torch.hub.download_url_to_file('https://github.com/ultralytics/yolov5/raw/master
8
  torch.hub.download_url_to_file('https://raw.githubusercontent.com/WongKinYiu/yolov7/main/inference/images/image3.jpg', 'image3.jpg')
9
  torch.hub.download_url_to_file('https://github.com/ultralytics/yolov5/releases/download/v5.0/yolov5s.pt','yolov5s.pt')
10
 
11
- model_path = "yolov5s.pt" #"yolov5m.pt", "yolov5l.pt", "yolov5x.pt",
12
  image_size = 640,
13
  conf_threshold = 0.25,
14
  iou_threshold = 0.45,
 
8
  torch.hub.download_url_to_file('https://raw.githubusercontent.com/WongKinYiu/yolov7/main/inference/images/image3.jpg', 'image3.jpg')
9
  torch.hub.download_url_to_file('https://github.com/ultralytics/yolov5/releases/download/v5.0/yolov5s.pt','yolov5s.pt')
10
 
11
+ model_path = "yolov5x.pt" #"yolov5s.pt" #"yolov5m.pt", "yolov5l.pt", "yolov5x.pt",
12
  image_size = 640,
13
  conf_threshold = 0.25,
14
  iou_threshold = 0.45,
app1.py CHANGED
@@ -8,7 +8,7 @@ torch.hub.download_url_to_file('https://github.com/ultralytics/yolov5/raw/master
8
  torch.hub.download_url_to_file('https://raw.githubusercontent.com/WongKinYiu/yolov7/main/inference/images/image3.jpg', 'image3.jpg')
9
  torch.hub.download_url_to_file('https://github.com/ultralytics/yolov5/releases/download/v5.0/yolov5s.pt','yolov5s.pt')
10
 
11
- model_path = "yolov5s.pt" #"yolov5m.pt", "yolov5l.pt", "yolov5x.pt",
12
  image_size = 640,
13
  conf_threshold = 0.25,
14
  iou_threshold = 0.45,
 
8
  torch.hub.download_url_to_file('https://raw.githubusercontent.com/WongKinYiu/yolov7/main/inference/images/image3.jpg', 'image3.jpg')
9
  torch.hub.download_url_to_file('https://github.com/ultralytics/yolov5/releases/download/v5.0/yolov5s.pt','yolov5s.pt')
10
 
11
+ model_path = "yolov5x.pt" #"yolov5s.pt" #"yolov5m.pt", "yolov5l.pt", "yolov5x.pt",
12
  image_size = 640,
13
  conf_threshold = 0.25,
14
  iou_threshold = 0.45,