Python get_meshrcnn_cfg_defaultsの例

コード例 #1

def setup_cfg(args):
    cfg = get_cfg()
    get_meshrcnn_cfg_defaults(cfg)
    cfg.merge_from_file(args.config_file)
    cfg.merge_from_list(args.opts)
    cfg.freeze()
    return cfg

コード例 #2

def setup_cfg(split_idx=0):
    splits = [
        "../meshrcnn/meshrcnn_R50.pth", "../meshrcnn/meshrcnn_S2_R50.pth"
    ]
    cfg = get_cfg()
    get_meshrcnn_cfg_defaults(cfg)
    cfg.merge_from_file("../meshrcnn/configs/viz.yaml")
    cfg.merge_from_list(["MODEL.WEIGHTS", splits[split_idx]])
    cfg.freeze()
    return cfg

コード例 #3

ファイル: train_net.py プロジェクト: peternara/meshrcnn-3d

def setup(args):
    cfg = get_cfg()
    get_meshrcnn_cfg_defaults(cfg)
    cfg.merge_from_file(args.config_file)
    cfg.merge_from_list(args.opts)
    cfg.freeze()
    default_setup(cfg, args)
    # Setup logger for "meshrcnn" module
    setup_logger(output=cfg.OUTPUT_DIR, distributed_rank=comm.get_rank(), name="meshrcnn")
    return cfg