Python LazyConfig.apply_overridesの例

コード例 #1

ファイル: benchmark.py プロジェクト: qilei123/detectron2

def setup(args):
    if args.config_file.endswith(".yaml"):
        cfg = get_cfg()
        cfg.merge_from_file(args.config_file)
        cfg.SOLVER.BASE_LR = 0.001  # Avoid NaNs. Not useful in this script anyway.
        cfg.merge_from_list(args.opts)
        cfg.freeze()
    else:
        cfg = LazyConfig.load(args.config_file)
        cfg = LazyConfig.apply_overrides(cfg, args.opts)
    setup_logger(distributed_rank=comm.get_rank())
    return cfg

コード例 #2

ファイル: lazyconfig_train_net.py プロジェクト: swhite2020/detectron2

def main(args):
    cfg = LazyConfig.load(args.config_file)
    cfg = LazyConfig.apply_overrides(cfg, args.opts)
    default_setup(cfg, args)

    if args.eval_only:
        model = instantiate(cfg.model)
        model = create_ddp_model(model)
        DetectionCheckpointer(model).load(cfg.train.init_checkpoint)
        print(do_test(cfg, model))
    else:
        do_train(args, cfg)

コード例 #3

ファイル: analyze_model.py プロジェクト: YangLiu14/detectron2-TAO

def setup(args):
    if args.config_file.endswith(".yaml"):
        cfg = get_cfg()
        cfg.merge_from_file(args.config_file)
        cfg.DATALOADER.NUM_WORKERS = 0
        cfg.merge_from_list(args.opts)
        cfg.freeze()
    else:
        cfg = LazyConfig.load(args.config_file)
        cfg = LazyConfig.apply_overrides(cfg, args.opts)
    setup_logger(name="fvcore")
    setup_logger()
    return cfg

コード例 #4

ファイル: test_lazy_config.py プロジェクト: SoumikCoder1/detectron2-MOT

 def test_invalid_overrides(self):
     cfg = LazyConfig.load(self.root_filename)
     with self.assertRaises(KeyError):
         LazyConfig.apply_overrides(cfg, ["lazyobj.x.xxx=123"])

コード例 #5

ファイル: test_lazy_config.py プロジェクト: SoumikCoder1/detectron2-MOT

 def test_overrides(self):
     cfg = LazyConfig.load(self.root_filename)
     LazyConfig.apply_overrides(cfg,
                                ["lazyobj.x=123", 'dir1b_dict.a="123"'])
     self.assertEqual(cfg.dir1b_dict.a, "123")
     self.assertEqual(cfg.lazyobj.x, 123)