def build_test_loader(cls, cfg, dataset_name):
if "YOLOF" == cfg.MODEL.META_ARCHITECTURE:
mapper = YOLOFDtasetMapper(cfg, False)
else:
mapper = None
return build_detection_test_loader(cfg, dataset_name, mapper=mapper)
def build_train_loader(cls, cfg):
if "YOLOF" == cfg.MODEL.META_ARCHITECTURE:
mapper = YOLOFDtasetMapper(cfg, True)
else:
mapper = None
return build_detection_train_loader(cfg, mapper=mapper)