Rename project files

habijung · Nov 28, 2021 · a67bfc8 · a67bfc8
1 parent f486965
commit a67bfc8
Show file tree

Hide file tree

Showing 2 changed files with 11 additions and 3 deletions.
diff --git a/run_extractor.py → run_mask_generator.py b/run_extractor.py → run_mask_generator.py
diff --git a/run_inpainting.py → run_video_inpainting.py b/run_inpainting.py → run_video_inpainting.py
@@ -368,6 +368,14 @@ def video_completion_seamless(args):
     # Loads masks.
     filename_list = glob.glob(os.path.join(args.path_mask, '*.png')) + \
                     glob.glob(os.path.join(args.path_mask, '*.jpg'))
+
+    if args.mode == 'square_removal':
+        tmp_list = []
+
+        for i in range(nFrame):
+            tmp_list.append(filename_list[0])
+
+        filename_list = tmp_list
 
     mask = []
     mask_dilated = []
@@ -609,8 +617,8 @@ def args_list(args):
 
 
 def main(args):
-    assert args.mode in ('object_removal', 'video_extrapolation'), (
-        "Accepted modes: 'object_removal', 'video_extrapolation', but input is %s"
+    assert args.mode in ('object_removal', 'square_removal'), (
+        "Accepted modes: 'object_removal', 'square_removal', but input is %s"
     ) % args.mode
 
     # Custom warnings
@@ -633,7 +641,7 @@ def main(args):
     parser = argparse.ArgumentParser()
 
     # video completion
-    parser.add_argument('--mode', default='object_removal', help="modes: object_removal / video_extrapolation")
+    parser.add_argument('--mode', default='object_removal', help="modes: object_removal / square_removal")
     parser.add_argument('--path', default='./data/color', help="dataset for evaluation")
     parser.add_argument('--path_mask', default='./data/mask', help="mask for object removal")
     parser.add_argument('--outroot', default='./data/result', help="output directory")