()
| 32 | |
| 33 | |
| 34 | def parse_args(): |
| 35 | parser = argparse.ArgumentParser(description='Organize MBench evaluation results') |
| 36 | parser.add_argument( |
| 37 | '--input_dir', |
| 38 | type=str, |
| 39 | required=True, |
| 40 | help='Input directory containing inference results' |
| 41 | ) |
| 42 | parser.add_argument( |
| 43 | '--output_dir', |
| 44 | type=str, |
| 45 | default='exp/mbench_eval_input', |
| 46 | help='Output directory for organized results' |
| 47 | ) |
| 48 | return parser.parse_args() |
| 49 | |
| 50 | |
| 51 | def load_motion_tensor(path: Path) -> torch.Tensor: |