backbone = efficientvit_backbone_l2(**kwargs)
if dataset == "cityscapes":
head = SegHead(
fid_list=["stage4", "stage3", "stage2"],
in_channel_list=[512, 256, 128],
stride_list=[32, 16, 8],
head_stride=8,
head_width=256,
head_depth=5,
expand_ratio=1,
middle_op="fmbconv",
final_expand=None,
n_classes=19,
act_func="gelu",
**build_kwargs_from_config(kwargs, SegHead)#在这里,build_kwargs_from_config 函数的目的是从 kwargs 中提取与 SegHead 类相关的参数,以便用于创建 SegHead 的实例。这样,可以在函数中轻松地配置 SegHead 对象的属性,而不需要显式指定每个参数。
,
)
build_kwargs_from_config
于 2023-10-10 13:23:40 首次发布