THP2903 commited on
Commit
1f82907
·
verified ·
1 Parent(s): b5476f5

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
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()