From a67bfc8643e695414c2a4ba4f96480a7acba6152 Mon Sep 17 00:00:00 2001 From: HabiJung Date: Sun, 28 Nov 2021 10:13:11 +0900 Subject: [PATCH] Rename project files --- run_extractor.py => run_mask_generator.py | 0 run_inpainting.py => run_video_inpainting.py | 14 +++++++++++--- 2 files changed, 11 insertions(+), 3 deletions(-) rename run_extractor.py => run_mask_generator.py (100%) rename run_inpainting.py => run_video_inpainting.py (98%) diff --git a/run_extractor.py b/run_mask_generator.py similarity index 100% rename from run_extractor.py rename to run_mask_generator.py diff --git a/run_inpainting.py b/run_video_inpainting.py similarity index 98% rename from run_inpainting.py rename to run_video_inpainting.py index c78a883..ae3a238 100644 --- a/run_inpainting.py +++ b/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")