()
| 12 | |
| 13 | |
| 14 | def parse_args(): |
| 15 | parser = argparse.ArgumentParser(description='VFdepth evaluation script') |
| 16 | parser.add_argument('--config_file', default ='./configs/surround_fusion.yaml', type=str, help='Config yaml file') |
| 17 | parser.add_argument('--weight_path', default = None, type=str, help='Pretrained weight path') |
| 18 | args = parser.parse_args() |
| 19 | return args |
| 20 | |
| 21 | |
| 22 | def test(cfg): |