(cfg)
| 125 | |
| 126 | |
| 127 | def build_test_loader(cfg): |
| 128 | |
| 129 | dataset_dicts = get_test_data_dicts(cfg.VISDRONE.TEST_JSON, cfg.VISDRONE.TEST_IMG_ROOT) |
| 130 | |
| 131 | dataset = DatasetFromList(dataset_dicts) |
| 132 | mapper = Mapper(cfg, False) |
| 133 | dataset = MapDataset(dataset, mapper) |
| 134 | |
| 135 | sampler = samplers.InferenceSampler(len(dataset)) |
| 136 | batch_sampler = torch.utils.data.sampler.BatchSampler(sampler, 1, drop_last=False) |
| 137 | |
| 138 | data_loader = torch.utils.data.DataLoader( |
| 139 | dataset, |
| 140 | num_workers=cfg.DATALOADER.NUM_WORKERS, |
| 141 | batch_sampler=batch_sampler, |
| 142 | collate_fn=trivial_batch_collator, |
| 143 | ) |
| 144 | return data_loader |
| 145 | |
| 146 | |
| 147 | def worker_init_reset_seed(worker_id): |
no test coverage detected