| 27 | |
| 28 | |
| 29 | def parse_args(): |
| 30 | parser = argparse.ArgumentParser() |
| 31 | parser.add_argument("--config") |
| 32 | parser.add_argument("--video_root", type=str) |
| 33 | parser.add_argument("--pose_root", type=str) |
| 34 | parser.add_argument("--ref_pose_root", type=str) |
| 35 | parser.add_argument("--source_mask_root", type=str) |
| 36 | parser.add_argument("--target_mask_root", type=str) |
| 37 | parser.add_argument("-W", type=int, default=512) |
| 38 | parser.add_argument("-H", type=int, default=512) |
| 39 | parser.add_argument("-L", type=int, default=24) |
| 40 | parser.add_argument("--seed", type=int, default=42) |
| 41 | parser.add_argument("--cfg", type=float, default=3.5) |
| 42 | parser.add_argument("--steps", type=int, default=50) |
| 43 | parser.add_argument("--fps", type=int, default=8) |
| 44 | parser.add_argument("--accumulation_steps", type=int, default=1) |
| 45 | parser.add_argument("--enable_xformers_memory_efficient_attention", type=str, default="Ture") |
| 46 | parser.add_argument("--gradient_checkpointing", type=str, default="True") |
| 47 | parser.add_argument("--num_inv_steps", type=int, default=50) |
| 48 | parser.add_argument("--suffix", type=str, default="jpg") |
| 49 | parser.add_argument("--camera", type=str, default="False") |
| 50 | |
| 51 | args = parser.parse_args() |
| 52 | |
| 53 | return args |
| 54 | |
| 55 | |
| 56 | def main(): |