def build_train_loader(cls, cfg):
mapper = DatasetMapper(cfg, True)
mapper.tfm_gens = cls.build_transform_gen(cfg, True)
return build_detection_train_loader(cfg, mapper=mapper)
def build_test_loader(cls, cfg, dataset_name):
mapper = DatasetMapper(cfg, False)
mapper.tfm_gens = cls.build_transform_gen(cfg, False)
return build_detection_test_loader(cfg, dataset_name, mapper=mapper)