Bugfix for batch demo: add padding if remaning number of frames is not equal to batch size

demo.sh

@@ -1,2 +1,2 @@
 #!/bin/sh
-python3 rtmo_demo.py ./video rtmo-t.fp16_batch1.onnx
+python3 rtmo_demo.py ./video rtmo-s.fp16_batch1.onnx

demo_batch.sh

@@ -1,2 +1,2 @@
 #!/bin/sh
-python3 rtmo_demo_batch.py ./video rtmo-t.fp16_batch4.onnx 4
+python3 rtmo_demo_batch.py ./video rtmo-s.fp16_batch4.onnx 4

rtmo_demo_batch.py

@@ -42,6 +42,14 @@ def process_video(video_path, body_estimator, batch_size=4):
 
     # Process remaining frames if any
     if batch_frames:
+        # Padding
+        while len(batch_frames) < batch_size:
+            # Option 1: Add a black frame
+            # black_frame = np.zeros_like(batch_frames[0])
+            # batch_frames.append(black_frame)
+
+            # Option 2: Duplicate the last frame
+            batch_frames.append(batch_frames[-1])
         batch_keypoints, batch_scores = body_estimator(batch_frames)
         for i, keypoints in enumerate(batch_keypoints):
             scores = batch_scores[i]