| 286 | |
| 287 | |
| 288 | def parse_args(): |
| 289 | parser = argparse.ArgumentParser(description='Inferencing') |
| 290 | parser.add_argument('--config', default='./configs/inference.yaml') |
| 291 | parser.add_argument('--name', default='test') |
| 292 | parser.add_argument('--from_dataset', action='store_true') |
| 293 | parser.add_argument('--cross_id', action='store_true') |
| 294 | parser.add_argument('--image_source', type=str, default=None, help='Single path or directory') |
| 295 | parser.add_argument('--video_source', type=str, default=None, help='Single path or directory') |
| 296 | parser.add_argument('--output_dir', default='./') |
| 297 | parser.add_argument('--inversion_option', type=str, default='encode', help='load, optimize, encode') |
| 298 | parser.add_argument('--frame_limit', type=int, default=1000) |
| 299 | parser.add_argument('--batch_size', type=int, default=10) |
| 300 | parser.add_argument('--if_extract', action='store_true') |
| 301 | parser.add_argument('--if_align', action='store_true') |
| 302 | # Audio |
| 303 | parser.add_argument('--enable_audio', action='store_true') |
| 304 | parser.add_argument('--audio_path', type=str, default=None) |
| 305 | # Editing |
| 306 | parser.add_argument('--attribute_edit', action='store_true') |
| 307 | parser.add_argument('--attribute', type=str, default=None) |
| 308 | parser.add_argument('--intuitive_edit', action='store_true') |
| 309 | parser.add_argument('--edit_expression_only', action='store_true') |
| 310 | args = parser.parse_args() |
| 311 | return args |
| 312 | |
| 313 | |
| 314 | def main(): |