Spaces:
Sleeping
Sleeping
Update app.py
Browse files
app.py
CHANGED
@@ -83,7 +83,7 @@ def image_predict(image):
|
|
83 |
res = ""
|
84 |
model_path = "best.pt"
|
85 |
model = YOLO(model_path)
|
86 |
-
results = model.predict(image,conf = 0.
|
87 |
box = results[0].boxes
|
88 |
res = check_acc(box)
|
89 |
annotated_frame = results[0].plot()
|
@@ -112,7 +112,7 @@ def extract_frames(video):
|
|
112 |
if (frame_no % (int(fps / nof))) == 0:
|
113 |
model_path = "best.pt"
|
114 |
model = YOLO(model_path)
|
115 |
-
results = model.predict(image,conf = 0.
|
116 |
box = results[0].boxes
|
117 |
res = check_acc(box)
|
118 |
annotated_frame = results[0].plot()
|
|
|
83 |
res = ""
|
84 |
model_path = "best.pt"
|
85 |
model = YOLO(model_path)
|
86 |
+
results = model.predict(image,conf = 0.7,iou = 0.4,imgsz = 512)
|
87 |
box = results[0].boxes
|
88 |
res = check_acc(box)
|
89 |
annotated_frame = results[0].plot()
|
|
|
112 |
if (frame_no % (int(fps / nof))) == 0:
|
113 |
model_path = "best.pt"
|
114 |
model = YOLO(model_path)
|
115 |
+
results = model.predict(image,conf = 0.7,iou = 0.4,imgsz = 512)
|
116 |
box = results[0].boxes
|
117 |
res = check_acc(box)
|
118 |
annotated_frame = results[0].plot()
|