Update MEMTrack/src/GenerateVideo.py

MEMTrack/src/GenerateVideo.py

@@ -12,8 +12,9 @@ def create_video(data_dir, image_dir, video_name,fps):
     height, width, channels = img_sample.shape
     
     video = cv2.VideoWriter(data_dir + video_name + ".mp4", fourcc, fps, (width, height))
-    #data_dir = "./Data/video3/"
-    #image_dir = os.path.join(_dir, "images")
+    print("Image dir: ", image_dir)
+    print("Number of frames: ", len(os.listdir(image_dir)))
+    print("Video output fps: ", fps)
     for frame in natsorted(os.listdir(image_dir)):
         #print(frame)
         img = cv2.imread(os.path.join(image_dir, frame))