Spaces:
Sleeping
Sleeping
Update app.py
#1
by
Mudassir-75
- opened
app.py
CHANGED
@@ -112,12 +112,13 @@ def process_video():
|
|
112 |
|
113 |
for frame_index, frame in enumerate(frames):
|
114 |
input_tensor = tf.image.resize_with_pad(tf.convert_to_tensor(frame, dtype=tf.uint8), 256, 256)
|
115 |
-
input_tensor = tf.cast(input_tensor, dtype=tf.float32
|
116 |
input_tensor = tf.expand_dims(input_tensor, axis=0)
|
117 |
keypoints = movenet_model.signatures['serving_default'](input_tensor)
|
118 |
keypoints_3d = keypoints['output_0'][0].numpy().tolist() # Assuming the model returns 3D keypoints
|
119 |
movenet_results.append(keypoints_3d)
|
120 |
|
|
|
121 |
# Detect stance based on keypoints (using ankles and wrists)
|
122 |
left_ankle = keypoints_3d[KEYPOINT_DICT['left_ankle']]
|
123 |
right_ankle = keypoints_3d[KEYPOINT_DICT['right_ankle']]
|
|
|
112 |
|
113 |
for frame_index, frame in enumerate(frames):
|
114 |
input_tensor = tf.image.resize_with_pad(tf.convert_to_tensor(frame, dtype=tf.uint8), 256, 256)
|
115 |
+
input_tensor = tf.cast(input_tensor, dtype=tf.int32) # Cast to int32 instead of float32
|
116 |
input_tensor = tf.expand_dims(input_tensor, axis=0)
|
117 |
keypoints = movenet_model.signatures['serving_default'](input_tensor)
|
118 |
keypoints_3d = keypoints['output_0'][0].numpy().tolist() # Assuming the model returns 3D keypoints
|
119 |
movenet_results.append(keypoints_3d)
|
120 |
|
121 |
+
|
122 |
# Detect stance based on keypoints (using ankles and wrists)
|
123 |
left_ankle = keypoints_3d[KEYPOINT_DICT['left_ankle']]
|
124 |
right_ankle = keypoints_3d[KEYPOINT_DICT['right_ankle']]
|