def build_detection_train_loader(cls, cfg, *args, mapper=None, **kwargs): logger.info("Building detection train loader ...") mapper = mapper or cls.get_mapper(cfg, is_train=True) logger.info("Using dataset mapper:\n{}".format(mapper)) sampler_name = cfg.DATALOADER.SAMPLER_TRAIN if sampler_name == "WeightedTrainingSampler": data_loader = build_weighted_detection_train_loader(cfg, mapper=mapper) else: data_loader = d2_build_detection_train_loader( cfg, *args, mapper=mapper, **kwargs ) if comm.is_main_process(): tbx_writer = _get_tbx_writer(get_tensorboard_log_dir(cfg.OUTPUT_DIR)) data_loader = cls.get_data_loader_vis_wrapper()( cfg, tbx_writer, data_loader ) return data_loader
def build_detection_train_loader(cls, *args, **kwargs): return d2_build_detection_train_loader(*args, **kwargs)