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 |
|