StevenGrove / DynamicHead

Apache License 2.0
86 stars 8 forks source link

请问dynamic_backbone在什么cfg中会去构建它? #6

Open tjmannn opened 3 years ago

tjmannn commented 3 years ago

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
StevenGrove commented 3 years ago

可以参考下https://github.com/Megvii-BaseDetection/cvpods/tree/master/playground/semantic_segmentation/cityscapes/dynamic_routing