(cfg)
| 82 | |
| 83 | |
| 84 | def build_test_loader(cfg): |
| 85 | transforms = build_transforms(cfg, is_train=False) |
| 86 | gallery_set = build_dataset(cfg.INPUT.DATASET, cfg.INPUT.DATA_ROOT, transforms, "gallery") |
| 87 | query_set = build_dataset(cfg.INPUT.DATASET, cfg.INPUT.DATA_ROOT, transforms, "query") |
| 88 | gallery_loader = torch.utils.data.DataLoader( |
| 89 | gallery_set, |
| 90 | batch_size=cfg.INPUT.BATCH_SIZE_TEST, |
| 91 | shuffle=False, |
| 92 | num_workers=cfg.INPUT.NUM_WORKERS_TEST, |
| 93 | pin_memory=True, |
| 94 | collate_fn=collate_fn, |
| 95 | ) |
| 96 | query_loader = torch.utils.data.DataLoader( |
| 97 | query_set, |
| 98 | batch_size=cfg.INPUT.BATCH_SIZE_TEST, |
| 99 | shuffle=False, |
| 100 | num_workers=cfg.INPUT.NUM_WORKERS_TEST, |
| 101 | pin_memory=True, |
| 102 | collate_fn=collate_fn, |
| 103 | ) |
| 104 | return gallery_loader, query_loader |
no test coverage detected