Update app.py
Browse files
app.py
CHANGED
@@ -12,7 +12,7 @@ from ultralytics import YOLO
|
|
12 |
#model_path = "yolov5x.pt" #"yolov5x.pt" "yolov5m.pt", "yolov5l.pt", "yolov5x.pt",
|
13 |
image_size = 640 #640 #320,
|
14 |
conf_threshold = 0.25 #0.30,
|
15 |
-
iou_threshold = 0.
|
16 |
#model = yolov5.load(model_path, device="cpu")
|
17 |
model = YOLO("yolo11x.pt") #YOLO("yolo11x.pt") #YOLO("yolo11x.pt")
|
18 |
|
|
|
12 |
#model_path = "yolov5x.pt" #"yolov5x.pt" "yolov5m.pt", "yolov5l.pt", "yolov5x.pt",
|
13 |
image_size = 640 #640 #320,
|
14 |
conf_threshold = 0.25 #0.30,
|
15 |
+
iou_threshold = 0.15
|
16 |
#model = yolov5.load(model_path, device="cpu")
|
17 |
model = YOLO("yolo11x.pt") #YOLO("yolo11x.pt") #YOLO("yolo11x.pt")
|
18 |
|