output

main.py

@@ -39,7 +39,7 @@ def check_extension(video):
     file_name = split_tup[0]
     file_extension = split_tup[1]
     print(file_extension)
-    
+
     if file_extension is not ".mp4":
         clip = moviepy.VideoFileClip(video)
 
@@ -88,11 +88,11 @@ def show_tracking(video_content):
         # Go through frames and write them 
         for frame_track in video_track:
             result_track = frame_track[0].plot()  # plot a BGR numpy array of predictions
-        
+            out_track.write(result_track)
+
         print("[INFO] Done with frames")
         #print(type(result_pose)) numpy ndarray
-        out_track.write(result_track)
-
+    
         out_track.release()
 
         video.release()