(args, split='train')
| 235 | return len(os.listdir(self.flows_dir)) |
| 236 | |
| 237 | def fetch_dataloader(args, split='train'): |
| 238 | print(colored('[Data]: ', 'yellow') + 'Loading MegaDepth CAPS') |
| 239 | dataset = MegaDepth_CAPS(args, split) |
| 240 | |
| 241 | if args.debug_data: |
| 242 | dataloader = data.DataLoader( |
| 243 | dataset, |
| 244 | batch_size=args.batch_size, |
| 245 | pin_memory=False, |
| 246 | shuffle=True, |
| 247 | num_workers=args.num_workers, |
| 248 | drop_last=True) |
| 249 | else: |
| 250 | sampler = DistributedSampler(dataset) |
| 251 | dataloader = data.DataLoader( |
| 252 | dataset, |
| 253 | batch_size=args.batch_size, |
| 254 | sampler=sampler) |
| 255 | |
| 256 | return dataloader |
| 257 | |
| 258 | |
| 259 | if __name__=='__main__': |
no test coverage detected