| 61 | |
| 62 | |
| 63 | def parse_args(): |
| 64 | parser = argparse.ArgumentParser() |
| 65 | parser.add_argument( |
| 66 | '-rmd', '--root_mask_dir', |
| 67 | type=str, |
| 68 | help="Root directory for mask_dirs" |
| 69 | ) |
| 70 | parser.add_argument( |
| 71 | '-rgd', '--root_gt_dir', |
| 72 | type=str, |
| 73 | help="Root directory for ground truth_dirs" |
| 74 | ) |
| 75 | parser.add_argument( |
| 76 | '-rrd', '--root_result_dir', |
| 77 | type=str, |
| 78 | help="Root directory for result_dirs" |
| 79 | ) |
| 80 | parser.add_argument( |
| 81 | '-mn', '--model_name', |
| 82 | type=str, |
| 83 | ) |
| 84 | parser.add_argument( |
| 85 | '-e', '--epoch', |
| 86 | type=str, |
| 87 | ) |
| 88 | parser.add_argument( |
| 89 | '-o', '--output_filename', |
| 90 | type=str, |
| 91 | ) |
| 92 | parser.add_argument( |
| 93 | '-rp', '--result_postfix', |
| 94 | type=str, |
| 95 | default='', |
| 96 | help="Result dir post dirname" |
| 97 | ) |
| 98 | parser.add_argument( |
| 99 | '-go', '--global_output_file', |
| 100 | type=str, |
| 101 | default="../../../data/checkpoints", |
| 102 | help="quantitative result log" |
| 103 | ) |
| 104 | parser.add_argument( |
| 105 | '-fc', '--flownet_checkpoint', |
| 106 | type=str, |
| 107 | default="/project/project-mira3/yaliangchang/FlowNet2_checkpoint.pth.tar", |
| 108 | help="Path to flownet2 checkpoint" |
| 109 | ) |
| 110 | parser.add_argument( |
| 111 | '-tn', '--test_num', |
| 112 | type=int, |
| 113 | default=100, |
| 114 | help="Number of videos to infer" |
| 115 | ) |
| 116 | parser.add_argument( |
| 117 | '--only_eval_fid', action='store_true', |
| 118 | help="Set this to evaluate only fid score" |
| 119 | ) |
| 120 | args = parser.parse_args() |