(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,
text_dir: Optional[Union[str, None]] = None,
token_dir: Optional[Union[str, None]] = None,
eval_cfg: Optional[Union[dict, None]] = None,
test_mode: Optional[bool] = False)
| 24 | text_dir (str): Path to the directory containing the text files. |
| 25 | """ |
| 26 | def __init__(self, |
| 27 | data_prefix: str, |
| 28 | pipeline: list, |
| 29 | dataset_name: Optional[Union[str, None]] = None, |
| 30 | fixed_length: Optional[Union[int, None]] = None, |
| 31 | ann_file: Optional[Union[str, None]] = None, |
| 32 | motion_dir: Optional[Union[str, None]] = None, |
| 33 | text_dir: Optional[Union[str, None]] = None, |
| 34 | token_dir: Optional[Union[str, None]] = None, |
| 35 | eval_cfg: Optional[Union[dict, None]] = None, |
| 36 | test_mode: Optional[bool] = False): |
| 37 | self.text_dir = os.path.join(data_prefix, 'datasets', dataset_name, text_dir) |
| 38 | if token_dir is not None: |
| 39 | self.token_dir = os.path.join(data_prefix, 'datasets', dataset_name, token_dir) |
| 40 | else: |
| 41 | self.token_dir = None |
| 42 | super(TextMotionDataset, self).__init__( |
| 43 | data_prefix=data_prefix, |
| 44 | pipeline=pipeline, |
| 45 | dataset_name=dataset_name, |
| 46 | fixed_length=fixed_length, |
| 47 | ann_file=ann_file, |
| 48 | motion_dir=motion_dir, |
| 49 | eval_cfg=eval_cfg, |
| 50 | test_mode=test_mode) |
| 51 | |
| 52 | def load_anno(self, name): |
| 53 | results = super().load_anno(name) |
nothing calls this directly
no outgoing calls
no test coverage detected