| 32 | |
| 33 | |
| 34 | def define_args(): |
| 35 | parser = argparse.ArgumentParser(description='3D lane evaluation') |
| 36 | # Paths settings |
| 37 | parser.add_argument('--dataset_dir', type=str, help='The path saving actual data') |
| 38 | parser.add_argument('--pred_dir', type=str, help='The path of prediction result') |
| 39 | parser.add_argument('--test_list', type=str, help='The path of test list txt') |
| 40 | # parser.add_argument('--images_dir', type=str, help='The path saving dataset images') |
| 41 | |
| 42 | return parser |
| 43 | |
| 44 | |
| 45 | def prune_3d_lane_by_visibility(lane_3d, visibility): |