コード例 #1
0
 def build_train_loader(cls, cfg: CfgNode):
     return build_detection_train_loader(cfg,
                                         mapper=DatasetMapper(cfg, True))
コード例 #2
0
 def build_test_loader(cls, cfg: CfgNode, dataset_name):
     return build_detection_test_loader(cfg,
                                        dataset_name,
                                        mapper=DatasetMapper(cfg, False))