Spaces:
Sleeping
Sleeping
Update app.py
Browse files
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
|
| 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
|
| 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):
|