Update visualization.py

visualization.py

@@ -270,6 +270,9 @@ def create_video_with_heatmap(video_path, df, mse_embeddings, mse_posture, outpu
         heatmap_img = heatmap_img.reshape(canvas.get_width_height()[::-1] + (3,))
         heatmap_img = cv2.resize(heatmap_img, (width, 200))
         
+        # Convert heatmap_img from RGB to BGR
+        heatmap_img = cv2.cvtColor(heatmap_img, cv2.COLOR_RGB2BGR)
+        
         combined_frame = np.vstack((frame, heatmap_img))
         
         seconds = frame_count / original_fps