Mauricio Guerta
commited on
Commit
·
3fd22b0
1
Parent(s):
4364716
Inicio
Browse files
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,
|