【扒网络架构】model.py

    # 自定义
    model = build_model(args).to(device)

    print(model)
    # 查看模型名称和大小
    for name, param in model.named_parameters():
        print(name, param.size())
    # 查看模块的参数数量
    param_count = sum(p.numel() for p in model.parameters())
    print(f"Total number of parameters of loca: {param_count}")

layer1 

layer2 

layer3 

layer4 

ccff conv1  conv2

bottlesnecks (0): RepVggBlock 

 bottlesnecks (1)、(2):RepVggBlock 

 (input_proj)

(encoder): TransformerEncoder

(ope): OPEModule

 (shape_or_objectness)

(regression_head): DensityMapRegressor

(aux_heads): ModuleList