Update MEMTrack/src/GenerateVideo.py

MEMTrack/src/GenerateVideo.py

@@ -9,11 +9,14 @@ def create_video(data_dir, image_dir, video_name,fps, all_image_path=None, all_i
     #print(data_dir)
     plotted_images = natsorted(os.listdir(image_dir))
     images = plotted_images
+    sample_image_dir = image_dir
     if all_images:
         all_images_list = natsorted(os.listdir(all_image_path))
         images = all_images_list
+        sample_image_dir = all_image_path
         
-    img_sample = cv2.imread(os.path.join(image_dir,images[0]))
+        
+    img_sample = cv2.imread(os.path.join(sample_image_dir,images[0]))
     print(images[0])
     print(img_sample.shape)
     height, width, channels = img_sample.shape