(self, aug_params=None, sparse=False)
| 17 | |
| 18 | class FlowDataset(data.Dataset): |
| 19 | def __init__(self, aug_params=None, sparse=False): |
| 20 | self.augmentor = None |
| 21 | self.sparse = sparse |
| 22 | if aug_params is not None: |
| 23 | if sparse: |
| 24 | self.augmentor = SparseFlowAugmentor(**aug_params) |
| 25 | else: |
| 26 | self.augmentor = FlowAugmentor(**aug_params) |
| 27 | |
| 28 | self.is_test = False |
| 29 | self.init_seed = False |
| 30 | self.flow_list = [] |
| 31 | self.image_list = [] |
| 32 | self.extra_info = [] |
| 33 | |
| 34 | def __getitem__(self, index): |
| 35 |
nothing calls this directly
no test coverage detected