Initialize MBench evaluator Args: device: Device to use for evaluation ('cuda' or 'cpu') output_path: Directory to save evaluation results full_info_dir: Path to full info JSON file (optional)
(self, device: str, output_path: str, full_info_dir: Optional[str] = None)
| 16 | """ |
| 17 | |
| 18 | def __init__(self, device: str, output_path: str, full_info_dir: Optional[str] = None): |
| 19 | """ |
| 20 | Initialize MBench evaluator |
| 21 | |
| 22 | Args: |
| 23 | device: Device to use for evaluation ('cuda' or 'cpu') |
| 24 | output_path: Directory to save evaluation results |
| 25 | full_info_dir: Path to full info JSON file (optional) |
| 26 | """ |
| 27 | self.device = device |
| 28 | self.output_path = output_path |
| 29 | self.full_info_dir = full_info_dir |
| 30 | os.makedirs(self.output_path, exist_ok=True) |
| 31 | self.category_dir_map = { |
| 32 | "motion_quality": "Motion_Quality", |
| 33 | "pose_quality": "Pose_Quality", |
| 34 | "motion_condition_consistency": "Motion_Condition_Consistency", |
| 35 | "motion_generalizability": "Motion_Generalizability" |
| 36 | } |
| 37 | |
| 38 | def build_full_dimension_list(self) -> List[str]: |
| 39 | """ |
nothing calls this directly
no outgoing calls
no test coverage detected