Spaces:
Sleeping
Sleeping
Update app.py
#3
by
Mudassir-75
- opened
app.py
CHANGED
@@ -92,7 +92,7 @@ def process_video():
|
|
92 |
# Extract 60 frames from the video
|
93 |
success, frame = cap.read()
|
94 |
frame_count = 0
|
95 |
-
while success and frame_count <
|
96 |
frames.append(frame)
|
97 |
success, frame = cap.read()
|
98 |
frame_count += 1
|
@@ -115,7 +115,7 @@ def process_video():
|
|
115 |
initial_right_wrist = None
|
116 |
|
117 |
for frame_index, frame in enumerate(frames):
|
118 |
-
input_tensor = tf.image.resize_with_pad(tf.convert_to_tensor(frame, dtype=tf.uint8),
|
119 |
input_tensor = tf.cast(input_tensor, dtype=tf.int32) # Cast to int32 instead of float32
|
120 |
input_tensor = tf.expand_dims(input_tensor, axis=0)
|
121 |
keypoints = movenet_model.signatures['serving_default'](input_tensor)
|
|
|
92 |
# Extract 60 frames from the video
|
93 |
success, frame = cap.read()
|
94 |
frame_count = 0
|
95 |
+
while success and frame_count < 20:
|
96 |
frames.append(frame)
|
97 |
success, frame = cap.read()
|
98 |
frame_count += 1
|
|
|
115 |
initial_right_wrist = None
|
116 |
|
117 |
for frame_index, frame in enumerate(frames):
|
118 |
+
input_tensor = tf.image.resize_with_pad(tf.convert_to_tensor(frame, dtype=tf.uint8), 128, 128)
|
119 |
input_tensor = tf.cast(input_tensor, dtype=tf.int32) # Cast to int32 instead of float32
|
120 |
input_tensor = tf.expand_dims(input_tensor, axis=0)
|
121 |
keypoints = movenet_model.signatures['serving_default'](input_tensor)
|