(self,
data_prefix: str,
pipeline: list,
dataset_name: Optional[Union[str, None]] = None,
fixed_length: Optional[Union[int, None]] = None,
ann_file: Optional[Union[str, None]] = None,
motion_dir: Optional[Union[str, None]] = None,
eval_cfg: Optional[Union[dict, None]] = None,
test_mode: Optional[bool] = False)
| 27 | """ |
| 28 | |
| 29 | def __init__(self, |
| 30 | data_prefix: str, |
| 31 | pipeline: list, |
| 32 | dataset_name: Optional[Union[str, None]] = None, |
| 33 | fixed_length: Optional[Union[int, None]] = None, |
| 34 | ann_file: Optional[Union[str, None]] = None, |
| 35 | motion_dir: Optional[Union[str, None]] = None, |
| 36 | eval_cfg: Optional[Union[dict, None]] = None, |
| 37 | test_mode: Optional[bool] = False): |
| 38 | super(BaseMotionDataset, self).__init__() |
| 39 | |
| 40 | self.data_prefix = data_prefix |
| 41 | self.pipeline = Compose(pipeline) |
| 42 | self.dataset_name = dataset_name |
| 43 | self.fixed_length = fixed_length |
| 44 | self.ann_file = os.path.join(data_prefix, 'datasets', dataset_name, ann_file) |
| 45 | self.motion_dir = os.path.join(data_prefix, 'datasets', dataset_name, motion_dir) |
| 46 | self.eval_cfg = copy.deepcopy(eval_cfg) |
| 47 | self.test_mode = test_mode |
| 48 | |
| 49 | self.load_annotations() |
| 50 | if self.test_mode: |
| 51 | self.prepare_evaluation() |
| 52 | |
| 53 | def load_anno(self, name): |
| 54 | motion_path = os.path.join(self.motion_dir, name + '.npy') |
nothing calls this directly
no test coverage detected