DynamicHead
DynamicHead copied to clipboard
请问dynamic_backbone在什么cfg中会去构建它?
cvpods/modeling/backbone/dynamic_arch/dynamic_backbone.py这个文件中有DynamicNetwork以及build_dynamic_backbone这个函数,请问哪个cfg文件或者这个repo中的什么检测器会去构建这个backbone?
这个函数下面的很多的参数设置有没有什么推荐值?
def build_dynamic_backbone(cfg, input_shape: ShapeSpec):
"""
Create a Dynamic Backbone from config.
Args:
cfg: a dl_lib CfgNode
Returns:
backbone (Backbone): backbone module, must be a subclass of :class:`Backbone`.
"""
if input_shape is None:
input_shape = ShapeSpec(channels=len(cfg.MODEL.PIXEL_MEAN))
backbone = DynamicNetwork(
init_channel=cfg.MODEL.BACKBONE.INIT_CHANNEL,
input_shape=input_shape,
cell_num_list=cfg.MODEL.BACKBONE.CELL_NUM_LIST,
layer_num=cfg.MODEL.BACKBONE.LAYER_NUM,
norm=cfg.MODEL.BACKBONE.NORM,
cal_flops=cfg.MODEL.CAL_FLOPS,
cell_type=cfg.MODEL.BACKBONE.CELL_TYPE,
max_stride=cfg.MODEL.BACKBONE.MAX_STRIDE,
sep_stem=cfg.MODEL.BACKBONE.SEPT_STEM,
using_gate=cfg.MODEL.GATE.GATE_ON,
small_gate=cfg.MODEL.GATE.SMALL_GATE,
gate_bias=cfg.MODEL.GATE.GATE_INIT_BIAS,
drop_prob=cfg.MODEL.BACKBONE.DROP_PROB
)
return backbone