(cls, cfg)
| 27 | |
| 28 | @classmethod |
| 29 | def build_train_loader(cls, cfg): |
| 30 | if "mix" in cfg.DATASETS.TRAIN[0]: |
| 31 | assert all(["mix" in t for t in cfg.DATASETS.TRAIN]), "Joint dataset only accepts only mix format" |
| 32 | mapper = MOTCenterNetDatasetMapper(cfg, is_train=True) |
| 33 | return build_mix_train_loader(cfg, mapper=mapper) |
| 34 | elif "mot" in cfg.DATASETS.TRAIN[0]: |
| 35 | mapper = MOTCenterNetDatasetMapper(cfg, is_train=True) |
| 36 | return build_mot_train_loader(cfg, mapper=mapper) |
| 37 | else: |
| 38 | mapper = COCOCenterNetDatasetMapper(cfg, is_train=True) |
| 39 | return build_detection_train_loader(cfg, mapper=mapper) |
| 40 | |
| 41 | @classmethod |
| 42 | def build_test_loader(cls, cfg, dataset_name): |
nothing calls this directly
no test coverage detected