(args, image_set)
| 264 | |
| 265 | |
| 266 | def build_dataset2transform(args, image_set): |
| 267 | mot17_train = make_transforms_for_mot17('train', args) |
| 268 | mot17_test = make_transforms_for_mot17('val', args) |
| 269 | |
| 270 | crowdhuman_train = make_transforms_for_crowdhuman('train', args) |
| 271 | dataset2transform_train = {'MOT17': mot17_train, 'CrowdHuman': crowdhuman_train} |
| 272 | dataset2transform_val = {'MOT17': mot17_test, 'CrowdHuman': mot17_test} |
| 273 | if image_set == 'train': |
| 274 | return dataset2transform_train |
| 275 | elif image_set == 'val': |
| 276 | return dataset2transform_val |
| 277 | else: |
| 278 | raise NotImplementedError() |
| 279 | |
| 280 | |
| 281 | def build(image_set, args): |
no test coverage detected