Mudassir-75 commited on
Commit
636df58
·
verified ·
1 Parent(s): 4b51408

Update app.py

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