Merge branch 'fabiofelix:main' into main

VIDA-NYU · Nov 16, 2024 · 0a2d0f8 · 0a2d0f8
2 parents 7ba0bef + 0421da1
commit 0a2d0f8
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 2 deletions.
diff --git a/step_recog/full/statemachine.py b/step_recog/full/statemachine.py
@@ -57,6 +57,8 @@ def smooth_probs(self, probabilities, type = SmoothType.MEDIAN):
         for pb in probs[1:]:
           move_avg[step] = self.ema_alpha * pb + (1 - self.ema_alpha) * move_avg[step]
 
+    #forces sum(move_avg) = 1.0
+    move_avg    = move_avg / np.sum(move_avg)
     max_prob    = np.max(move_avg)
     max_indices = np.where(move_avg == max_prob)[0]
 

diff --git a/step_recog/full/visualize.py b/step_recog/full/visualize.py
@@ -46,8 +46,7 @@ def main(video_path, output_path='output.mp4', cfg_file=""):
               psm.process_timestep(prob_step)
               step_idx  = np.argmax(prob_step)
               step_desc = "No step" if step_idx >= len(model.STEPS) else model.STEPS[step_idx]   
-
-            pbar.set_description(" ".join(f"{x:.0%}" for x in prob_step) + " | " + " ".join(f'{x}' for x in psm.current_state))            
+              pbar.set_description(" ".join(f"{x:.0%}" for x in prob_step) + " | " + " ".join(f'{x}' for x in psm.current_state))            
 
             # draw the prediction (could be your bar chart) on the frame
             plot_graph(frame, prob_step, step_desc, psm.current_state[:-1])