add datetime to output files to prevent conflict

FGT_codes/tool/video_inpainting.py

@@ -776,7 +776,7 @@ def video_inpainting(args, imgArr, imgMaskArr):
     for i in range(len(comp_frames)):
         comp_frames[i] = comp_frames[i].astype(np.uint8)
     imageio.mimwrite(
-        os.path.join(args.outroot, "result.mp4"), comp_frames, fps=args.out_fps, quality=8
+        os.path.join(args.outroot, args.outfilename), comp_frames, fps=args.out_fps, quality=8
     )
     print(f"Done, please check your result in {args.outroot} ")
 

app.py

@@ -12,6 +12,7 @@ import sys
 from os.path import exists, join, basename, splitext
 import os
 import argparse
+from datetime import datetime
 
 project_name = ''
 
@@ -46,6 +47,11 @@ siammask = Custom(anchors=cfg['anchors'])
 siammask = load_pretrain(siammask, pretrained_path1)
 siammask = siammask.eval().to(device)
 
+# datetime object containing current date and time
+now = datetime.now()
+ 
+dt_string = now.strftime("%d_%m_%Y_%H_%M_%S")
+print("date and time =", dt_string)
 # constants
 object_x = 0
 object_y = 0
@@ -69,6 +75,8 @@ parser.add_argument(
     '--path_mask', default='/myData/dilateAnnotations_4/walking', help="mask for object removal")
 parser.add_argument(
     '--outroot', default=os.path.abspath(project_name), help="output directory")
+parser.add_argument(
+    '--outfilename', default=dt_string+"_result.mp4", help="output filename")
 parser.add_argument('--consistencyThres', dest='consistencyThres', default=5, type=float,
                     help='flow consistency error threshold')
 parser.add_argument('--alpha', dest='alpha', default=0.1, type=float)
@@ -156,7 +164,7 @@ def track_and_mask(vid, original_frame, masked_frame):
         in_fps = fps
         # can't write out mp4, so try to write into an AVI file
         video_writer = cv2.VideoWriter(
-            "output.avi", cv2.VideoWriter_fourcc(*'MP42'), fps, (width, height))
+            dt_string+"_output.avi", cv2.VideoWriter_fourcc(*'MP42'), fps, (width, height))
 
         while video_capture.isOpened():
             ret, frame = video_capture.read()
@@ -198,14 +206,14 @@ def track_and_mask(vid, original_frame, masked_frame):
 
     print('Original Frame Count: ',len(original_frame_list))
     print('Mask Frame Count: ',len(mask_list))
-    return "output.avi"
+    return dt_string+"_output.avi"
 
 
 def inpaint_video():
     args.out_fps = in_fps
     video_inpainting(args, original_frame_list, mask_list)
 
-    return "result.mp4"
+    return dt_string+"_result.mp4"
 
 
 def get_first_frame(video):