咱们做视觉检测的同学,这两年应该没少被一个问题折腾:怎么把一辆车的“前后脸”快速准确地认出来。我最近正好把一个完整任务跑通了——用YOLO11做汽车保险杠前后位置识别,顺带整理了一套保险杠位置识别数据集,把标注文件也按YOLO格式洗干净了。这篇就把整个链路从头到尾聊一遍:数据怎么做、标注文件什么格式、参数怎么调、训练完了怎么判断好坏,还有我踩过的几个比较隐蔽的坑。内容不挑基础,只要你用过一点Python、碰过标注工具,就能照着把保险杠检测这个任务从头跑通。
这个任务看着简单,但真做起来会发现前后保险杠在不少车型上长得非常像,尤其现在新能源车流行封闭式前脸和隐藏式排气,单靠“有没有进气格栅”来判断前后根本不靠谱。所以数据质量、标注规则和类别设计就变得比模型本身还重要。我会把这一整套经验拆开讲,适合正在做车辆外观检测、自动泊车辅助、车辆定损、配件识别这类方向的朋友参考。
1. 保险杠位置识别到底在解决什么问题
1.1 业务场景拆解:你可能会在哪些地方用到这个能力
保险杠检测不是一个新的算法题,但它在实际业务里出现的频率比想象中高得多。我接触到的需求大致可以分三类。
第一类是车辆外观损伤检测。停车场出入口、无人洗车机、二手车检测平台,都需要判断车辆前后保险杠是否有划痕、裂纹或变形。这类业务不只要求知道“这里有保险杠”,还要求知道是前保险杠还是后保险杠,因为施工流程和定损单价完全不同。前保补漆和后保换件的工艺、工时、配件编号都对不上,位置认错就全乱了套。
第二类是自动泊车和辅助驾驶的局部感知。泊车系统需要知道前后保险杠的轮廓范围,用来估算车辆与障碍物的距离,或者判断车位线是否被车身自身遮挡。这类场景不用像分割那样精细,但检测框的稳定性要求很高,视频流里不能来回跳。
第三类是配件适配和改装审核。电商平台或汽修系统拍一张车辆前后部照片,需要自动判断出车型对应的保险杠款式,以及是否加装了前唇、后包围这类改装配件。这类任务通常会拿保险杠检测结果去做二次分类,所以框得准不准直接决定下游准确率。
我之前看过一个项目,最初用普通的车辆检测模型再裁切子图,结果在侧方停车视角下经常把前保和后保同时裁进一个框里,精度根本提不上去。后来改成端到端识别“前保险杠”和“后保险杠”两个独立类别,效果立刻就稳定了。这个经验很关键:如果业务最终需要区分前后,就一定不要只检测“保险杠”一个类别,把前后分开标注,等于把分类信息直接教给了模型。
1.2 为什么选YOLO11而不是继续用旧版本
选择YOLO11不是因为它新,而是因为这个任务本身对模型有几点硬性要求,YOLO11恰好都能满足。
第一点是对中小目标的召回能力。车辆停在远处时,保险杠在画面里可能只有几十个像素高,尤其是后保险杠,视觉特征本来就弱:纯色、低纹理、和车身几乎融为一体。YOLO11在骨干网络里换掉了一部分旧的C2f结构,用了新的C3k2模块,下采样过程中对小目标的语义保留效果比老一代好,我在同样数据下对比过YOLOv8和YOLO11的召回率,大概能高两到三个百分点。
第二点是训练和推理的灵活性。YOLO11和之前的Ultralytics系列一样,一条命令行就能从训练走到导出,还支持OpenVINO、ONNX、TensorRT这些推理后端。像我们这种经常要换部署环境的团队,最怕的就是训练一套、部署又要改一套,YOLO11这套生态能省掉很多桥接工作。
第三点是类别可扩展性。车辆检测任务往往会从单纯检测保险杠扩展到检测大灯、尾灯、车牌、后视镜等多个部件。YOLO11的检测头还是解耦的类别预测,新增类别只需要改data.yaml和标注文件,不需要动网络结构,后面做多部件检测非常方便。
当然,如果你手头已经有一套跑得很顺的YOLOv8管道,也不用非迁移不可。但如果是新任务起步,我会直接推荐用YOLO11,毕竟旧项目迁移的成本不会太大,新项目没必要一上来就用老架构。
需要模型API调用? 免费领10W Token,多模型网关一键接入 Claude、DeepSeek 等主流模型。
2. 保险杠数据集的构建:从采集到一份能用的YOLO标注文件
2.1 数据采集的取舍标准:不是照片越多越好
做保险杠位置识别,数据的采集思路和通用目标检测不太一样。通用检测要的是“见多识广”,保险杠检测要的是“角度和距离受控”。因为业务里最后用到的视角基本就那几种:正前方、正后方、左右45度、俯视(洗车机/举升机视角)。你拍一堆侧面大远景,模型学完在正前方视角反而不稳。
我整理数据集的时候,按下面几个维度来控制采集比例:
- 视角:正前、正后各占35%,左右45度各占15%,俯视占15%。
- 车距:近景占50%,中景占35%,远景占15%。远景样本太少模型容易漏检小尺寸保险杠。
- 车型:轿车、SUV、MPV、皮卡都要有,新能源车至少占三成。现在很多新车型前保险杠和大灯做成一体贯穿式,后保险杠也几乎没有排气管口,视觉特征和传统燃油车差很多,样本不够就等着泛化崩。
- 光照:白天正常光占60%,逆光和阴影各占15%,夜间/车库光占10%。如果业务场景在地下停车场,那夜间样本比例还要往上加。
采集来源比较常见的是:园区停车场、4S店维修工位、道路监控卡口、共享出行平台的验车照片。条件允许的话可以拿仿真引擎渲染一批带标注的合成数据,用来补充极端角度,但要注意合成数据和真实照片之间风格差距大,我会控制在不超过总量20%。
这里有个实操细节:原始视频可以先抽帧,但尽量隔几帧抽一张,避免连续帧高度相似造成训练集和验证集数据泄漏,也就是俗称的“近亲样本”。如果同一辆车从同一视角连续抽了50帧,模型看着训练集指标很高,验证集又是同一辆车,mAP全虚高,换一批真实数据立刻打回原形。
2.2 标注规则与YOLO标注文件格式:前后分开才能用得明白
标注规则是整个任务里最容易被低估的一环。我见过很多合作方拿分割掩膜标完整个保险杠,然后问怎么转成YOLO检测框。实际上YOLO检测标注只需要外接矩形,不需要像素级轮廓。矩形框要保证两点:一是完整包住保险杠本体,二是尽量贴合边缘,不要包进去一大块无关的翼子板或机盖。
在类别划分上,我的建议是用两个类:
0:front_bumper(前保险杠)1:rear_bumper(后保险杠)
标注的时候有一个容易犯的错误:有些照片里车前脸有明显的镀铬条或进气格栅,标注员就只框格栅那一块。这不对,保险杠的物理边界一般是延伸到轮拱前缘的整个下部包围区域。所以训练前最好让标注员统一看20张示例图,把“框哪里”这个问题在开工前解决掉,否则返工成本非常大。
标注工具用labelme或者labelImg都行。LabelImg可以直接导出YOLO格式,但如果团队习惯用labelme,它导出的是JSON文件,需要自己写一个转换脚本。转出来的YOLO标注文件格式是这样的:
code复制0 0.512 0.731 0.321 0.154
1 0.488 0.704 0.295 0.163
每行五个数字,含义依次是:类别ID、中心点x坐标(相对图片宽度归一化)、中心点y坐标(相对图片高度归一化)、框宽度(归一化)、框高度(归一化)。数值范围0到1,不需要对应具体像素,这是YOLO格式和COCO格式最大的区别,不少人刚开始会在这里绕一下。
我建议前期就把像素坐标转归一化坐标的脚本写好。一个典型的转换脚本逻辑如下:
python复制import json
import os
def convert_labelme_to_yolo(json_path, target_dir, class_map):
with open(json_path, 'r', encoding='utf-8') as f:
data = json.load(f)
img_w = data['imageWidth']
img_h = data['imageHeight']
lines = []
for shape in data['shapes']:
label = shape['label']
if label not in class_map:
continue
points = shape['points']
xs = [p[0] for p in points]
ys = [p[1] for p in points]
x_min, x_max = min(xs), max(xs)
y_min, y_max = min(ys), max(ys)
box_w = x_max - x_min
box_h = y_max - y_min
x_center = (x_min + x_max) / 2.0 / img_w
y_center = (y_min + y_max) / 2.0 / img_h
norm_w = box_w / img_w
norm_h = box_h / img_h
lines.append(f"{class_map[label]} {x_center:.6f} {y_center:.6f} {norm_w:.6f} {norm_h:.6f}")
base_name = os.path.splitext(os.path.basename(json_path))[0]
with open(os.path.join(target_dir, base_name + '.txt'), 'w') as f:
f.write('\n'.join(lines))
所有图片放到同一目录下,所有txt放到另一个目录,保持同名,这就是YOLO训练时默认的对应方式。
最终数据集目录结构长这样:
code复制bumper_dataset/
├── images/
│ ├── train/
│ ├── val/
│ └── test/
├── labels/
│ ├── train/
│ ├── val/
│ └── test/
└── bumper.yaml
其中data.yaml的写法也固定:
yaml复制path: /path/to/bumper_dataset
train: images/train
val: images/val
test: images/test
nc: 2
names:
0: front_bumper
1: rear_bumper
注意path这一项最好写绝对路径,YOLO在训练时会基于这个路径拼接train和val的相对路径。如果写成相对路径,换一台机器跑就很容易报数据集路径找不到。
3. 训练前的最后一步:环境配置与模型选型
3.1 软硬件环境:显存不够也能跑
训练保险杠检测这种任务,对硬件的要求比通用检测低不少。因为输入尺寸通常640就够用,类别只有两个,模型复杂度不需要拉满。我自己这次是用一张比较普通的消费级显卡跑的,显存12G,batch调到16完全没问题。
如果你手头的卡显存不大,比如只有8G甚至6G,也不是不能跑,把batch降到8或4,然后适当减少mosaic开启的概率就行。再不行就选YOLO11n这种最小尺寸的权重,照样能跑出一个工程可用的模型。
软件方面,我建议用干净的虚拟环境,Python版本3.9到3.11之间都可以。然后只装一个核心依赖:
bash复制pip install ultralytics
Ultralytics这个库会自动把torch、torchvision一起拉进来,不需要先单独装PyTorch。但如果你机器上已经有特定版本的PyTorch了,建议先确认版本兼容性再装,避免把环境搞乱。
装完可以快速检查一下安装是否正常:
bash复制python -c "from ultralytics import YOLO; print('ok')"
正常会打印ok,如果卡在库导入阶段,先排查是不是OpenCV版本冲突。
3.2 YOLO11五个尺寸选哪个:小模型不见得差
YOLO11提供了n/s/m/l/x五个尺寸,参数量和速度差别很大。保险杠检测这种任务,我不建议一上来就选最大的x版本。原因很简单:类别少、目标结构相对简单,大模型带来的提升有限,但训练时间和推理延迟却会实打实翻好几倍。
五个尺寸大概可以参考这个表格:
| 模型尺寸 | 参数量 | 特点 | 适用情况 |
|---|---|---|---|
| yolo11n | 约2.6M | 速度最快,精度相对低 | 边缘设备、实时性要求高 |
| yolo11s | 约9.4M | 性价比高 | 大多数常规项目首选 |
| yolo11m | 约20.1M | 精度和速度均衡 | 精度要求较高、算力够用 |
| yolo11l | 约25.3M | 精度更好 | 离线分析、精度优先 |
| yolo11x | 约56.9M | 精度最强 | 算力充足、追求极致精度 |
我这次最后选了s,训练约120个epoch,验证集mAP50能到0.96左右,mAP50-95大概0.88。这个成绩已经足够支撑下游的分类逻辑了。如果部署平台是Jetson这种边缘设备,直接用n,损失一点精度,但能保证实时。
选好尺寸之后,可以直接在命令行里指定预训练权重:
bash复制yolo detect train \
model=yolo11s.pt \
data=bumper.yaml \
epochs=150 \
imgsz=640 \
batch=16 \
device=0 \
project=runs/bumper \
name=train_v1
第一次运行会自动下载对应的预训练权重,存放在当前目录的weights文件夹里。如果你网络不好,也可以手动下载后在model参数里指定本地路径。
3.3 关键训练超参数:哪些值得调,哪些用默认
刚开始跑保险杠检测的时候,我建议大部分超参数保持默认,只动几个关键的。等模型能正常收敛了,再针对问题点调。
epochs:先设100到150。保险杠检测任务收敛速度不算慢,100轮通常已经能看到平台期。如果训练loss还在明显下降,可以再加。imgsz:默认640。如果业务图里保险杠占比很小,可以提高到1280,但显存占用会涨很多,先不要贸然开。batch:根据显存来。尽量设大,大batch能让BN层统计更稳定。12G显存跑s模型,16没问题。patience:早停机制,默认50。意思是50轮没有更好结果就自动停。我建议保持默认,这样不用死等固定epoch数。workers:数据加载线程数,默认8。Windows下如果报DataLoader相关错误,可以改成2再试。device:指定GPU编号,单卡就是0;CPU跑虽然能跑通但非常慢,只适合验证流程。
另外有两个和保险杠场景直接相关的增强选项值得留意:hsv_h、hsv_s这类颜色增强默认就是开的,对白天夜间光强差异大的场景有好处;flipud垂直翻转只建议在俯视样本占比较高的场景开,一般场景默认不开或者只开很小的概率,不然会把正前方保险杠翻成“倒挂在天空上”的奇怪样本。
4. 训练过程:怎么看曲线,怎么判断模型真的能用了
4.1 训练日志里那些指标到底什么意思
训练过程中,终端会滚动输出一列指标,很多人只盯mAP,其实前期的关键指标有两个:box_loss和cls_loss。
box_loss是框回归的误差,保险杠位置识别对框的位置精度要求高,这个loss要持续下降。cls_loss是分类误差,也就是能不能分清前保和后保。两类loss如果都在正常下降,说明训练在往前走。
当训练结束后,Ultralytics会在runs/bumper/train_v1/里生成五个关键指标,重点看这几个:
results.png:综合曲线图,包含loss曲线和mAP曲线。confusion_matrix.png:混淆矩阵,能直接看到前后保险杠有没有互相认错。PR_curve.png:PR曲线,曲线下的面积越大越好。val_batch0_pred.jpg:验证集预测结果可视化,需要人工检查框贴不贴边、类别对不对。
混淆矩阵是我每次必看的图。如果front_back对角线很亮,说明类别分得开。如果font和rear那一块有比较亮的块面,也就是常说的“前后互混”,那就不是单纯调参数的问题,要去检查标注类别是否标错,或者某些车型前后脸确实太像,需要补充对应车型的样本。
4.2 训练时的几个异常信号
训练保险杠检测的时候,我遇到过三种比较典型的异常,大家可以对照着排查。
第一种是loss降不下去,卡在某个平台期。通常是数据里有不少难样本,比如远处模糊的保险杠、被遮阳板挡住的保险杠。这种情况先不要加模型复杂度,而是把难样本单独抽出来看看标注是否标全了。有时候一张图里明明有三个保险杠,只标了两个,模型被错误的背景信息反复教育,loss就迟迟不降。
第二种是训练loss正常降,但验证mAP忽高忽低。这往往是验证集和训练集分布差异大,或者数据量太少。数据量不够最狠的做法是把验证集比例调低到10%,或者直接用K折交叉验证。如果连折交叉验证的机器都带不动,那就得回去采集数据。
第三种是mAP50很高,但mAP50-95上不去。说明模型框得不够精准,预测框和真值框的重合度一般在0.5到0.75之间。病根通常是标注框太松,边缘出入好几个像素。保险杠边缘往往是一条渐变过渡带,想让mAP50-95上去,重新精修标注比换模型更有效。
我在第二次跑的时候就把验证集里那些框太松的标注全部重新标了一遍,单精修了一百多张,mAP50-95从0.80直接涨到0.88。这个收益比把s换m大得多,成本也低。
4.3 前保险杠和后保险杠在训练策略上要不要区别对待
一个很有用的技巧是:前保险杠和后保险杠在特征复杂度上不太一样,数据增强的比例可以区别处理。
前保险杠一般特征丰富,有格栅、雾灯、车牌位、雷达开孔,模型通常学得快。后保险杠特征相对单一,很多车型后保就是一块光秃秃的塑料覆盖件,再加上有些车漆颜色和保险杠颜色几乎一致,模型学起来慢多了。
我的做法是把后保险杠占比低的图片做一次离线增强:对后保险杠区域做局部亮度扰动和轻微模糊,生成一批补充样本。注意不要整张图大幅改变颜色,否则前保险杠也跟着变,相当于数据集扩容了但没有针对性。局部扰动后,后保险杠的召回率大概能提高3到5个点。
另外,夜间样本少的情况下,可以试一下YOLO11自带的augment透传,在后处理阶段混入随机Gamma变换,比手工改亮度泛化能力好一些。
5. 推理部署与前后辨识的工程细节
5.1 推理流程:从单张图片到视频流
训练完得到的best.pt就可以直接用来做推理。单张图片测试最简单:
python复制from ultralytics import YOLO
model = YOLO('runs/bumper/train_v1/weights/best.pt')
results = model('test.jpg', conf=0.4, iou=0.5)
for r in results:
boxes = r.boxes
for box in boxes:
cls_id = int(box.cls[0])
conf = float(box.conf[0])
x1, y1, x2, y2 = box.xyxy[0].tolist()
label = model.names[cls_id]
print(label, conf, x1, y1, x2, y2)
这里有个经验:做保险杠检测时置信度阈值不要设太低。我一般用0.35到0.45之间。别小看这个阈值,保险杠的纹理本来就少,阈值设太低会导致车门、翼子板上的阴影被当成保险杠。反过来说,iou默认0.5不用动,如果你觉得同一个保险杠被重复框住很多次,可以适当调高到0.6。
视频流推理也简单,只要把source参数从图片路径换成视频文件路径或摄像头ID:
bash复制yolo detect predict \
model=runs/bumper/train_v1/weights/best.pt \
source=car_video.mp4 \
conf=0.4 \
save=True
如果在实际业务里追求吞吐量,我建议把模型导出为ONNX格式再跑。导出命令:
bash复制yolo export model=runs/bumper/train_v1/weights/best.pt format=onnx imgsz=640
导出的best.onnx可以用ONNX Runtime加载,推理速度比PyTorch原生快不少,而且不依赖PyTorch环境,C++端也能接。
ONNX Runtime加载示例:
python复制import cv2
import numpy as np
import onnxruntime as ort
session = ort.InferenceSession('best.onnx', providers=['CUDAExecutionProvider', 'CPUExecutionProvider'])
input_name = session.get_inputs()[0].name
input_shape = session.get_inputs()[0].shape
image = cv2.imread('test.jpg')
image = cv2.cvtColor(image, cv2.COLOR_BGR2RGB)
image = cv2.resize(image, (640, 640))
img = image.astype(np.float32) / 255.0
img = img.transpose(2, 0, 1)[None, ...]
outputs = session.run(None, {input_name: img})
ONNX输出的维度格式和PyTorch直接推理不完全一样,需要按640缩放比例把检测框换算回原图坐标。这里建议在导出后先拿同一张图对比PyTorch结果和ONNX结果的框坐标,确认转换正确再接入业务,这个对比步骤能省下后面一堆排查时间。
5.2 前后辨识的实用技巧:不只是等模型输出
虽然模型已经输出了front_bumper和rear_bumper类别,但在实际业务里还需要做一些额外的位置校验,因为单帧检测偶尔会出错。
第一个技巧是几何先验校验。如果画面里同时出现了车牌和保险杠,前保险杠必然和车牌在同一区域,且车牌在保险杠内部或上方;后保险杠则和尾灯、后车标位置绑定。检测出保险杠后,可以用小型的车牌检测结果做交叉验证:当检测框类型和车牌位置出现矛盾时,以置信度高的结果为准或丢弃这一帧。这个技巧在视频流里尤其好用,能滤掉不少闪烁误检。
第二个技巧是时序平滑。视频流里保险杠检测框偶尔会轻微抖动,直接计算距离时会很不稳。我习惯做一个轻量的指数移动平均,把每一帧的检测框中心坐标和历史中心坐标做加权平均。权重给0.7左右,效果就很柔和,又不会拖影太严重。
第三个技巧是利用“前后保险杠互斥”的天然物理约束。一辆车在画面里不可能既正面前方又正面后方,如果检测结果里同框出现大量front和rear且都有较高置信度,那这一帧大概率处于45度角视角,这时候可以保留面积较大的那个框,或者干脆降权处理,避免下游逻辑算错。
这些技巧看着简单,但在真实业务里能省下大量定位错误的问题。模型是基础,规则是兜底,两者配合才可靠。
6. 实测数据与踩坑记录
6.1 三种典型场景下的表现差异
我在自己整理的数据集上跑了三个测试场景:白天园区停车场、地下车库、雨后的湿滑路面。白天场景最稳,front_bumper的mAP50达到0.97,rear_bumper稍微低一点在0.95左右,主要是有几辆浅色车在浅色背景下后保险杠确实难认。地下车库的难点在灯光色温差异大,有的车位是暖黄光、有的冷白光,模型在冷白光下漏检率会升高,但不会高到不可用。雨后路面最麻烦,地面积水会倒影出保险杠,检测框偶尔会跳到倒影上,这时对检测框做Y坐标下限约束就很有用——保险杠不可能出现在低于轮胎接触面的位置。
如果和你业务场景接近,建议直接按照这个思路设计验证集,分别统计三个场景的mAP,只盯着整体mAP容易被平均成绩欺骗。
6.2 标注不规范引发的连锁问题
前前后后做了几轮保险杠数据,我最大的教训是标注一致性。第一轮数据质量差,主要问题有三个:有的人从格栅边缘就开始框,把机盖下沿包进去;有的人把保险杠和翼子板一刀切,导致框的一侧凹进去;还有的人把后保险杠反射条单独当成检测目标。这三个问题直接造成训练时正样本框的形状方差特别大,模型学出来的框忽大忽小。
后来我定了一套“保险杠标注三段式”规则,效果很直观:
- 左右边界要到轮拱前缘或后缘,不把轮胎包进去。
- 上边界到机盖/尾门与保险杠的接缝线。
- 下边界到保险杠本体下沿,不把底盘阴影算进去。
另外,每个人标注习惯不同,标注完成后我建议抽样检查至少15%的框,重点看边界是否统一。一个小的标注不一致,到模型上就是几个点的精度差,不值得。
6.3 几个容易被忽略的坑
第一个坑是中文路径问题。Ultralytics在Windows下如果数据集路径带中文,经常会出现读取失败或者写结果失败。我的建议是整个项目目录和数据集目录都用英文路径,省心。
第二个坑是预训练权重版本匹配。如果你不是从官方源下载的预训练权重,要留意它是不是和当前Ultralytics版本兼容。高版本的模型结构或者参数名变了,旧权重可能直接加载失败,或者加载完精度异常。稳妥的做法是:
bash复制pip list | grep ultralytics
看版本号,同时下载对应版本的权重文件。
第三个坑是ONNX导出的动态批处理。默认导出固定batch=1,对单图推理够用,但如果想跑多batch推理,需要加dynamic=True参数。保险杠检测一般单张处理就够,除非你的业务是批量跑整库图片。
第四个坑是夜间拍摄的防抖虚影。夜间快门慢,车辆稍微一动保险杠边缘就虚,标注员很难标准。我后来在采集夜间样本时用手持稳定器或者固定机位拍摄,数据质量明显提升。
一些想对你说的实在话
做保险杠识别这个任务,模型部分其实不复杂,真正决定项目上限的是数据设计和工程细节。YOLO11本身是个很成熟的目标检测框架,你不必花太多时间在调网络结构上,把精力放到数据标注规范、前后类别平衡、难样本补充这些更实在的地方,回报会高很多。
如果你也正在做类似的项目,我特别建议提前做好数据集版本管理,哪怕只是按日期归档命名。保险杠数据的特点是车型更新换代快,半年前的模型很可能在下一批新车型照片上掉点。有版本管理就能快速定位是数据分布的问题还是模型的问题。
最后再说一个小技巧:训练结束后,把验证集里置信度最低的那批预测可视化导出来,一张张看。你会发现自己模型真正弱的到底是大灯造型特殊的车、还是深色车漆的车、还是遮挡严重的场景。这一步比盲目调参有用得多。我就是靠这个办法,把后保险杠的误检率又压下去了一大截。
