Ultralytics YOLO27: 加入候补名单。
获取 YOLO 许可
返回 Ultralytics 术语表

Backbone

主干网络是视觉模型底层的预训练网络,它将像素转换为多尺度特征图,供颈部网络和检测头使用。

主干网络是神经网络中将原始输入转换为供模型其余部分使用的特征图的部分。在计算机视觉中,主干网络通常是经过 ImageNet 等大型数据集预训练的卷积神经网络 (CNN)或视觉 Transformer (ViT)。其后的层通常包括颈部和任务专用的头部,会将特征转换为类别标签、边界框、掩码或关键点。主干网络、编码器和特征提取器通常可以互换使用;在 U-Net等编码器—解码器模型中,通常使用「编码器」这一名称。

主干网络的工作原理#

视觉主干网络分阶段处理图像。每个阶段都会应用一组卷积或注意力模块,然后将空间分辨率减半,并通常增加通道数;因此,640×640 的输入会变成步幅为 8 的 80×80 特征图、步幅为 16 的 40×40 特征图,以及步幅为 32 的 20×20 特征图。较早的阶段会响应边缘、颜色和纹理;较后的阶段则会响应物体部件和完整物体,因为这些阶段的每个输出位置都覆盖了输入图像中更大的感受野。检测和分割模型会从最后三个阶段提取特征,并将其传递给特征金字塔网络 (FPN)等颈部结构;这样,模型就能在高分辨率特征图中找到小物体,并在低分辨率特征图中找到大物体。

骨干网络、颈部与头部的区别#

大多数现代物体检测架构由三个部分组成;替换组件或冻结层时,这些部分之间的边界很重要。

  • 主干网络:负责特征提取,并在多个分辨率上生成特征图。主干网络包含可迁移到不同任务和数据集的通用视觉知识。
  • 颈部:融合主干网络不同尺度的特征,使每个输出层级同时保留精细的空间细节和高层语义。FPN 添加自上而下的路径,PANet 则在此基础上再添加一条自下而上的路径。分类模型通常没有颈部,而是将主干网络的最后一个阶段池化为单个向量。
  • 头部:将每个融合后的特征图转换为任务输出。检测头预测边界框和类别分数;分割头和姿态头则根据同类特征预测掩码和关键点。

常见的主干网络系列#

  • ResNet:残差连接让信号能够跳过某些模块,使多达 152 层的网络也能训练;ResNet 集成模型赢得了 ILSVRC 2015 分类任务。ResNet-50 至今仍是比较检测器时常用的标准参考主干网络。
  • CSPNet:跨阶段部分网络只将每个阶段的一部分通道送入该阶段的模块,并在末尾将其余通道合并回来;在 ImageNet 准确率持平或更高的情况下,这种设计将计算量降低了 20%。YOLOv4 的主干网络 CSPDarknet53 就由这些阶段构成。
  • ConvNeXt:采用从 Transformer 借鉴的设计选择(如大卷积核和更少的归一化层)改进 ResNet,在 ImageNet 上达到 87.8% 的 top-1 准确率,并在 COCO 检测任务中胜过 Swin Transformer。
  • ViT 和 Swin Transformer:标准 ViT 从头到尾都保持单一分辨率,因此检测模型通常改用 Swin;Swin 在移位的局部窗口内计算注意力,并生成类似 CNN 的分阶段层级结构。
  • 自监督基础主干网络:DINOv3在 17 亿张无标签图像上训练了一个拥有 70 亿参数的 ViT;冻结主干网络时,其特征在分割和深度估计等密集任务上仍能取得优异结果。

预训练主干网络与迁移学习#

从头训练骨干网络需要大型数据集和较长的训练周期,因此大多数项目会从预训练权重开始。这种迁移学习之所以有效,是因为在一个数据集上学到的边缘、纹理和形状特征对大多数其他数据集也有用。对于接近预训练领域的小型数据集,冻结骨干网络可以保留这些通用特征,减少过拟合并降低训练内存占用,因为冻结层不会存储梯度或优化器状态。对于大型数据集或差异较大的领域(例如X 射线或热成像),同时对骨干网络进行微调可以让其特征适应新领域。在 YOLO26中,骨干网络包含第 0 至 10 层(步幅卷积、C3k2 块、SPPF 池化块和 C2PSA 注意力块);freeze=11训练参数会冻结这些层,使训练仅更新颈部和检测头;微调指南会详细讲解这一配置。

from ultralytics import YOLO

# 加载在 COCO 上预训练的 YOLO26 检测模型
model = YOLO("yolo26n.pt")

# 冻结整个主干网络(第 0–10 层),在小型数据集上进行微调
results = model.train(data="coco8.yaml", epochs=3, imgsz=640, freeze=11)

第一个 epoch 开始前,训练器会记录 Freezing layer 'model.0.conv.weight',并为第 0–10 层中的其他每个参数记录对应内容;这些参数会保持固定,而颈部和头部则学习新数据。

探索解决方案

体育领域的计算机视觉

体育领域的计算机视觉

了解 Ultralytics YOLO 如何通过球员和球追踪、表现分析、教练洞察及运动员康复,推动体育视觉技术的发展。
了解详情
航空影像计算机视觉

航空影像计算机视觉

使用 Ultralytics YOLO 将无人机和航空影像转化为农业、水产养殖、保护工作和地理空间分析的实时洞察。
了解详情
航空航天中的计算机视觉

航空航天中的计算机视觉

使用 Ultralytics YOLO 将计算机视觉应用于航空航天领域,包括飞机检查、跑道 FOD 检测、机场地面作业和复合材料缺陷检测。
了解详情
安防领域的计算机视觉

安防领域的计算机视觉

使用 Ultralytics YOLO 模型守护每个场所。视觉 AI 可用于周界监控、威胁检测、车牌识别和工作场所安全。
了解详情
机器人领域的计算机视觉

机器人领域的计算机视觉

使用 Ultralytics YOLO 模型提升智能机器的能力。机器人领域的视觉 AI 可用于自主导航、感知、目标跟踪和实时控制。
了解详情
物流中的计算机视觉

物流中的计算机视觉

借助 Ultralytics YOLO 模型优化物流流程。视觉 AI 可用于包裹检查、分拣、车辆跟踪和仓库安全实时监控。
了解详情
零售中的计算机视觉

零售中的计算机视觉

借助 Ultralytics YOLO 模型重新构想零售。视觉 AI 可用于库存跟踪、货架监控、排队管理和更深入的客户洞察。
了解详情
医疗领域的计算机视觉

医疗领域的计算机视觉

借助 Ultralytics YOLO 模型构建医疗解决方案。医疗领域的视觉 AI 可加快医学影像处理、优化诊断并监测患者。
了解详情
制造业中的计算机视觉

制造业中的计算机视觉

利用 Ultralytics YOLO 模型优化制造流程。视觉 AI 可用于质量控制、缺陷检测、个人防护装备合规检查和装配线自动化。
了解详情
汽车领域的计算机视觉

汽车领域的计算机视觉

利用 Ultralytics YOLO 模型在汽车领域应用计算机视觉。视觉 AI 可提升道路安全、驾驶辅助和车辆自动化水平,让道路更智能。
了解详情
农业中的计算机视觉

农业中的计算机视觉

借助 Ultralytics YOLO 模型,将视觉 AI 引入智慧农业。为作物监测、牲畜追踪和精准农业提供支持,从而提高产量并实现更智能的生产。
了解详情

让我们携手构建 AI 的未来!

开启你的旅程,迎接机器学习的未来