imperiusrex commited on
Commit
ef376b2
·
verified ·
1 Parent(s): b45ccc7

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -30,13 +30,13 @@ def process_video(video_path):
30
  break
31
  results = extract_model(frame)
32
  labels = [extract_model.names[int(c)] for c in results[0].boxes.cls.cpu().numpy()]
33
- if "board" in labels and "person" not in labels:
34
  frames.append(frame)
35
  cv2.imwrite(f"frames/frame_{idx:04d}.jpg", frame)
36
  idx += 1
37
  cap.release()
38
  if not frames:
39
- raise RuntimeError("No frames with only 'board' and no 'person' found.")
40
 
41
  # Step 2: Align
42
  def align_frames(ref, tgt):
 
30
  break
31
  results = extract_model(frame)
32
  labels = [extract_model.names[int(c)] for c in results[0].boxes.cls.cpu().numpy()]
33
+ if "board" in labels:
34
  frames.append(frame)
35
  cv2.imwrite(f"frames/frame_{idx:04d}.jpg", frame)
36
  idx += 1
37
  cap.release()
38
  if not frames:
39
+ raise RuntimeError("No frames with 'board' found.")
40
 
41
  # Step 2: Align
42
  def align_frames(ref, tgt):