(self,
data_prefix: str,
pipeline: list,
dataset_name: str,
body_model: Optional[Union[dict, None]] = None,
ann_file: Optional[Union[str, None]] = None,
convention: Optional[str] = 'human_data',
cache_data_path: Optional[Union[str, None]] = None,
test_mode: Optional[bool] = False)
| 69 | } |
| 70 | |
| 71 | def __init__(self, |
| 72 | data_prefix: str, |
| 73 | pipeline: list, |
| 74 | dataset_name: str, |
| 75 | body_model: Optional[Union[dict, None]] = None, |
| 76 | ann_file: Optional[Union[str, None]] = None, |
| 77 | convention: Optional[str] = 'human_data', |
| 78 | cache_data_path: Optional[Union[str, None]] = None, |
| 79 | test_mode: Optional[bool] = False): |
| 80 | self.convention = convention |
| 81 | self.num_keypoints = get_keypoint_num(convention) |
| 82 | self.cache_data_path = cache_data_path |
| 83 | super(HumanImageDataset, |
| 84 | self).__init__(data_prefix, pipeline, ann_file, test_mode, |
| 85 | dataset_name) |
| 86 | if body_model is not None: |
| 87 | self.body_model = build_body_model(body_model) |
| 88 | else: |
| 89 | self.body_model = None |
| 90 | |
| 91 | def get_annotation_file(self): |
| 92 | """Get path of the annotation file.""" |
nothing calls this directly
no test coverage detected