From 62fd922646fe64112f327cbdeb97b432d37a690c Mon Sep 17 00:00:00 2001 From: Stupdi Go Date: Sun, 11 Jan 2026 15:48:44 -0600 Subject: [PATCH] AI, we don't believe that you can fly bro pt 3 --- training.py | 18 ++++++++++-------- 1 file changed, 10 insertions(+), 8 deletions(-) diff --git a/training.py b/training.py index 702c7f8..7c0a7c0 100644 --- a/training.py +++ b/training.py @@ -158,15 +158,17 @@ def get_features_sequence(landmarks_seq, max_frames=100): if len(landmarks_seq) > 1: deltas[1:] = landmarks_seq[1:] - landmarks_seq[:-1] - # Combine all features - seq = np.concatenate([ - landmarks_seq, # (T, 21, 3) - deltas, # (T, 21, 3) - curl_features[:, :, np.newaxis] # (T, 5, 1) - ], axis=2) + # Flatten each component separately, then concatenate + landmarks_flat = landmarks_seq.reshape(landmarks_seq.shape[0], -1) # (T, 63) + deltas_flat = deltas.reshape(deltas.shape[0], -1) # (T, 63) + # curl_features is already (T, 5) - # Flatten spatial dimensions: (T, 21*3 + 21*3 + 5) = (T, 131) - seq = seq.reshape(seq.shape[0], -1) + # Combine: 63 + 63 + 5 = 131 features per frame + seq = np.concatenate([ + landmarks_flat, + deltas_flat, + curl_features + ], axis=1) # Pad or truncate to max_frames T, F = seq.shape