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