def get_default_cfg():
"""
Override `get_default_cfg` for adding non common config.
"""
from detectron2.config import get_cfg as get_d2_cfg
cfg = get_d2_cfg()
cfg = CN(cfg) # upgrade from D2's CfgNode to D2Go's CfgNode
cfg.SOLVER.AUTO_SCALING_METHODS = ["default_scale_d2_configs"]
return cfg
def get_infer_topk_cfg():
cfg = get_d2_cfg()
cfg.TEST.TOPK_CAT = CfgNode()
cfg.TEST.TOPK_CAT.ENABLED = True
cfg.TEST.TOPK_CAT.K = 10000
cfg.TEST.TOPK_CAT.MIN_SCORE = 1.0e-7
# Images used to estimate initial score threshold, with mask branch off.
# Set to be greater than LVIS validation set, so we do a full pass with the
# mask branch off.
cfg.TEST.TOPK_CAT.NUM_ESTIMATE = 30000
return cfg
def get_default_cfg():
"""
Override `get_default_cfg` for adding non common config.
"""
from detectron2.config import get_cfg as get_d2_cfg
cfg = get_d2_cfg()
cfg = CfgNode.cast_from_other_class(
cfg) # upgrade from D2's CfgNode to D2Go's CfgNode
try:
from d2go.runner import get_unintentional_added_configs_during_runner_import
for key in get_unintentional_added_configs_during_runner_import():
cfg.register_deprecated_key(key)
except ImportError:
pass
cfg.SOLVER.AUTO_SCALING_METHODS = ["default_scale_d2_configs"]
return cfg
