(args, **kwargs)
| 6 | |
| 7 | |
| 8 | def build_dataset(args, **kwargs): |
| 9 | # images |
| 10 | if args.dataset == 'imagenet': |
| 11 | return build_imagenet(args, **kwargs) |
| 12 | if args.dataset == 'imagenet_code': |
| 13 | return build_imagenet_code(args, **kwargs) |
| 14 | if args.dataset == 'coco': |
| 15 | return build_coco(args, **kwargs) |
| 16 | if args.dataset == 'openimage': |
| 17 | return build_openimage(args, **kwargs) |
| 18 | if args.dataset == 'pexels': |
| 19 | return build_pexels(args, **kwargs) |
| 20 | if args.dataset == 't2i_image': |
| 21 | return build_t2i_image(args, **kwargs) |
| 22 | if args.dataset == 't2i': |
| 23 | return build_t2i(args, **kwargs) |
| 24 | if args.dataset == 't2i_code': |
| 25 | return build_t2i_code(args, **kwargs) |
| 26 | |
| 27 | raise ValueError(f'dataset {args.dataset} is not supported') |
no test coverage detected