Puyush commited on
Commit
ec19816
1 Parent(s): 1c9bdd1

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
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.6,iou = 0.3,imgsz = 512)
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.6,iou = 0.3,imgsz = 512)
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()