| 71 | return args |
| 72 | |
| 73 | def get_demo_video_args(): |
| 74 | parser = argparse.ArgumentParser() |
| 75 | parser.add_argument('--demo_root', type=str, default='./demo/demo_video/') |
| 76 | parser.add_argument('--scene_name', type=str, default='scene.mp4') |
| 77 | parser.add_argument('--marker_name', type=str, default='fantastic_beast.jpg') |
| 78 | parser.add_argument('--movie_name', type=str, default='movie.mp4') |
| 79 | parser.add_argument('--save_name', type=str, default='out.avi') |
| 80 | parser.add_argument('--movie_start_idx', type=int, default=0) |
| 81 | parser.add_argument('--scene_start_idx', type=int, default=150) |
| 82 | |
| 83 | parser.add_argument('--test', action='store_true', help='test specified scene and frame') |
| 84 | parser.add_argument('--scene_id', type=int, default=0) |
| 85 | parser.add_argument('--frame_id', type=int, default=0) |
| 86 | parser.add_argument('--draw', action='store_true') |
| 87 | parser.add_argument('--save', action='store_true') |
| 88 | |
| 89 | return parser.parse_args() |
| 90 | |
| 91 | def get_demo_harsh_lighting_args(): |
| 92 | parser = argparse.ArgumentParser() |