Spaces:
Sleeping
Sleeping
Update app.py
Browse files
app.py
CHANGED
|
@@ -41,10 +41,10 @@ def process_video_audio(video_path, audio_path):
|
|
| 41 |
(x, y, w, h) = faces[0]
|
| 42 |
face = frame[y:y+h, x:x+w]
|
| 43 |
resized_face = cv2.resize(face, (120, 120))
|
| 44 |
-
train_visual[0, :, :, :, frame_idx] = resized_face
|
| 45 |
else:
|
| 46 |
resized_frame = cv2.resize(frame, (120, 120))
|
| 47 |
-
train_visual[0, :, :, :, frame_idx] = resized_frame
|
| 48 |
last_frame = frame
|
| 49 |
frame_idx += 1
|
| 50 |
cap.release()
|
|
|
|
| 41 |
(x, y, w, h) = faces[0]
|
| 42 |
face = frame[y:y+h, x:x+w]
|
| 43 |
resized_face = cv2.resize(face, (120, 120))
|
| 44 |
+
train_visual[0, :, :, :, frame_idx] = pt.tensor(resized_face)
|
| 45 |
else:
|
| 46 |
resized_frame = cv2.resize(frame, (120, 120))
|
| 47 |
+
train_visual[0, :, :, :, frame_idx] = pt.tensor(resized_frame)
|
| 48 |
last_frame = frame
|
| 49 |
frame_idx += 1
|
| 50 |
cap.release()
|