| 2 | |
| 3 | |
| 4 | def config_parser(): |
| 5 | parser = configargparse.ArgumentParser() |
| 6 | # general |
| 7 | parser.add_argument("--config", is_config_file=True, help="config file path") |
| 8 | parser.add_argument( |
| 9 | "--rootdir", |
| 10 | type=str, |
| 11 | default="./", |
| 12 | help="the path to the project root directory. Replace this path with yours!", |
| 13 | ) |
| 14 | parser.add_argument("--expname", type=str, help="experiment name") |
| 15 | parser.add_argument("--distributed", action="store_true", help="if use distributed training") |
| 16 | parser.add_argument("--local_rank", type=int, default=0, help="rank for distributed training") |
| 17 | parser.add_argument( |
| 18 | "-j", |
| 19 | "--workers", |
| 20 | default=8, |
| 21 | type=int, |
| 22 | metavar="N", |
| 23 | help="number of data loading workers (default: 8)", |
| 24 | ) |
| 25 | |
| 26 | ########## dataset options ########## |
| 27 | ## train and eval dataset |
| 28 | parser.add_argument( |
| 29 | "--train_dataset", |
| 30 | type=str, |
| 31 | default="ibrnet_collected", |
| 32 | help="the training dataset, should either be a single dataset, " |
| 33 | 'or multiple datasets connected with "+", for example, ibrnet_collected+llff+spaces', |
| 34 | ) |
| 35 | parser.add_argument( |
| 36 | "--dataset_weights", |
| 37 | nargs="+", |
| 38 | type=float, |
| 39 | default=[], |
| 40 | help="the weights for training datasets, valid when multiple datasets are used.", |
| 41 | ) |
| 42 | parser.add_argument( |
| 43 | "--train_scenes", |
| 44 | nargs="+", |
| 45 | default=[], |
| 46 | help="optional, specify a subset of training scenes from training dataset", |
| 47 | ) |
| 48 | parser.add_argument( |
| 49 | "--eval_dataset", type=str, default="llff_test", help="the dataset to evaluate" |
| 50 | ) |
| 51 | parser.add_argument( |
| 52 | "--eval_scenes", |
| 53 | nargs="+", |
| 54 | default=[], |
| 55 | help="optional, specify a subset of scenes from eval_dataset to evaluate", |
| 56 | ) |
| 57 | ## others |
| 58 | parser.add_argument( |
| 59 | "--testskip", |
| 60 | type=int, |
| 61 | default=8, |