尧图网页设计 
简介本资源为基于Traffic-Net训练交通拥堵程度识别的完整项目包面向计算机、人工智能、数据科学等专业的在校学生与教师适合作为课程设计、大作业、毕业设计或入门深度学习实践的参考案例。压缩包共10个文件约6KB以py源码、json模型配置、md说明文档和txt提示文件为主其中Python脚本承载Traffic-Net网络训练与识别逻辑json文件记录模型类别等关键配置md与txt则提供运行说明和常见问题提示结构精简、便于快速理解项目脉络。目前已有50人学习下载。通过该资源读者可掌握交通拥堵程度识别的建模思路与训练流程理解模型配置与数据组织的对应关系并在此基础上进行二次开发或功能扩展积累深度学习项目落地的实践经验。1. 交通拥堵程度识别到底在识别什么从 Traffic-Net 的输入输出说起拿到「基于Traffic-Net训练交通拥堵程度识别源码数据集下载.zip」这个包多数人第一反应是解压、装依赖、跑 train.py。但真正决定你能不能复现出可用结果的不是代码写得多漂亮而是你搞清楚 Traffic-Net 这类模型到底吃什么、吐什么。交通拥堵程度识别本质是一个把道路图像或视频帧映射到拥堵等级的分类/回归任务输入通常是固定视角的卡口或路侧相机画面输出是畅通、缓行、拥堵、严重拥堵这类离散等级或者 0 到 1 之间的拥堵指数。它和车辆检测、车流量统计不是一回事检测关心「有几辆车、在哪」拥堵识别关心「这条路现在堵成什么样」后者对全局纹理、车辆密度、车间距、车道占用率的敏感度远高于单目标框的精度。这套源码加数据集的价值在于它把数据读取、模型定义、训练循环、评估指标打包好了你不需要从零搭一个拥堵分类 pipeline。适合谁做智慧交通课程设计的学生、需要快速验证路侧感知方案的算法工程师、以及手上有自己卡口数据想迁移训练的从业者。但要注意拥堵程度这个标签本身带有主观性不同标注员对「缓行」和「拥堵」的边界判断可能不一致这是你后面调参和看指标时必须心里有数的前提。热搜里常出现的 ccpd数据集、车辆检测数据集bdd100、自动驾驶数据集和拥堵识别是近邻但不同任务别混用标签体系。2. 把 Traffic-Net 的源码和数据集跑起来环境、目录与最小训练命令2.1 解压后先看目录结构别急着 pip install拿到 zip 之后我一般先 tree 一下看目录确认数据集和源码的相对路径。常见结构是Traffic-Net/下分datasets/、models/、train.py、config.yaml、requirements.txt。数据集可能是按类别分文件夹的 ImageFolder 格式也可能是 csv 标注加图片路径。先确认这一点因为后面 DataLoader 怎么写完全取决于它。# 查看压缩包内容不解压先看结构 unzip -l Traffic-Net训练交通拥堵程度识别源码数据集下载.zip | head -50 # 解压到指定目录 unzip Traffic-Net训练交通拥堵程度识别源码数据集下载.zip -d traffic_net_work # 查看目录树两层 cd traffic_net_work find . -maxdepth 2 -type d | sort逻辑说明unzip -l只列出内容不落盘避免解压出一堆乱码文件名后才发现结构不对。参数上-d指定解压目录建议单独建工作目录别在下载目录里直接跑训练否则缓存文件和权重会混在一起。如果压缩包内是中文文件名Linux 下可能乱码用unzip -O GBK或7z x处理。2.2 依赖安装与 CUDA 版本对齐requirements.txt 里通常有 torch、torchvision、numpy、opencv-python、pillow、tqdm、matplotlib。坑在于 torch 版本和 CUDA 驱动不匹配这是血泪经验里排第一的翻车点。# 建虚拟环境别污染系统 python python -m venv venv_traffic source venv_traffic/bin/activate # Windows 用 venv_traffic\Scripts\activate # 先看本机 CUDA 驱动支持的最高版本 nvidia-smi # 按官方对应关系装 torch这里以 cu118 为例按你驱动改 pip install torch torchvision --index-url https://download.pytorch.org/whl/cu118 # 再装其余依赖 pip install -r requirements.txt逻辑说明nvidia-smi右上角 CUDA Version 是驱动支持的上限不是你必须装的版本装低于它的即可。参数上--index-url指向 PyTorch 官方 wheel 源比默认 pypi 快且版本全。如果装完torch.cuda.is_available()返回 False先别怀疑代码九成是 torch 和驱动不匹配重装对应版本即可。2.3 数据集划分与最小训练命令拥堵识别数据集通常按 7:2:1 划分 train/val/test。如果源码没带划分脚本自己按类别分层抽样保证每个拥堵等级在三个集合里比例一致否则验证集指标会虚高。import os, random, shutil from pathlib import Path src Path(datasets/raw) # 原始按类别分好的目录 dst Path(datasets/split) ratios {train: 0.7, val: 0.2, test: 0.1} random.seed(42) for cls_dir in src.iterdir(): if not cls_dir.is_dir(): continue imgs list(cls_dir.glob(*.jpg)) list(cls_dir.glob(*.png)) random.shuffle(imgs) n len(imgs) n_train, n_val int(n * ratios[train]), int(n * ratios[val]) splits { train: imgs[:n_train], val: imgs[n_train:n_train n_val], test: imgs[n_train n_val:], } for split, files in splits.items(): out dst / split / cls_dir.name out.mkdir(parentsTrue, exist_okTrue) for f in files: shutil.copy(f, out / f.name) print(split done)逻辑说明按类别分别抽样再复制保证分层。random.seed(42)固定随机性方便复现。参数上 ratios 可按数据集大小调样本少于 500 张时 test 比例可降到 0.1 以下把更多数据留给训练。复制而非移动保留原始数据后悔药。# 最小训练命令具体参数名以源码 config 为准 python train.py --data datasets/split --epochs 50 --batch-size 32 --lr 1e-3 --img-size 224 --gpu 0逻辑说明--data指向划分后根目录--img-size要和模型输入一致Traffic-Net 常见输入是 224 或 256。--batch-size受显存限制8G 显存跑 224 输入一般 32 到 64。先跑 2 个 epoch 确认 loss 在降、没有 NaN再挂长训练。3. Traffic-Net 模型结构与拥堵等级分类头怎么改3.1 主干网络选型轻量还是精度优先Traffic-Net 这个名字在不同实现里主干可能是 ResNet、MobileNet、EfficientNet 或自研 CNN。路侧设备算力有限时MobileNetV3 或 ShuffleNet 更实际追求精度且用服务器推理ResNet50 起步。判断方法很简单看源码 models 目录里 import 的是哪个 torchvision 模型或者有没有自定义 backbone 文件。主干参数量级适用场景拥堵识别注意点MobileNetV3约 5M边缘设备、实时对纹理细节弱严重拥堵易混ResNet50约 25M服务器、离线精度稳显存占用高EfficientNet-B0约 5M折中方案输入分辨率敏感别乱改自研 CNN看实现特定数据集泛化差迁移需谨慎选型理由拥堵等级区分主要靠车辆密度和排列纹理不需要检测框级别细节所以轻量主干往往够用。但如果你数据集里「缓行」和「拥堵」视觉差异极小就得靠更深的主干或更高输入分辨率。3.2 分类头改造与类别数对齐源码默认类别数可能和你的数据集不一致这是必须改的第一处。假设你的数据集是 4 类拥堵等级而源码写死 3 类不改就报维度错。import torch.nn as nn from torchvision import models num_classes 4 # 畅通/缓行/拥堵/严重拥堵 model models.resnet50(weightsIMAGENET1K_V2) # 替换最后的全连接层 in_features model.fc.in_features model.fc nn.Sequential( nn.Dropout(0.3), nn.Linear(in_features, 256), nn.ReLU(inplaceTrue), nn.Linear(256, num_classes) ) print(model.fc)逻辑说明weightsIMAGENET1K_V2加载预训练权重拥堵数据集通常几千到几万张从头训容易过拟合迁移学习是标配。中间加一层 256 维和 Dropout是为了在小数据集上增加非线性同时抑制过拟合。参数上 Dropout 0.3 到 0.5 之间调数据越少取值越大。类别数一定和数据集文件夹数量一致否则训练不报错但评估全乱。3.3 损失函数与类别不平衡处理拥堵数据天然不平衡畅通样本远多于严重拥堵。直接用 CrossEntropyLoss 会让模型偏向多数类严重拥堵召回率极低。import numpy as np import torch # 统计各类样本数 class_counts np.array([1200, 800, 400, 150]) # 按你的实际统计填 weights 1.0 / class_counts weights weights / weights.sum() * len(class_counts) class_weights torch.tensor(weights, dtypetorch.float32) criterion nn.CrossEntropyLoss(weightclass_weights, label_smoothing0.1)逻辑说明1/类别数再归一化让少数类获得更大权重。label_smoothing0.1缓解标注主观性带来的噪声拥堵等级边界模糊时尤其有用。参数上权重别拉太极端否则少数类过拟合、多数类崩掉一般控制在 10 倍以内。4. 训练过程监控与拥堵识别指标怎么读4.1 该盯哪些曲线不该只看 accuracy拥堵识别里 accuracy 是最容易骗人的指标。四类均衡时 25% 是随机水平但如果畅通占 70%模型全预测畅通也有 70% accuracy实际毫无用处。必须同时看每类的 precision、recall、F1以及混淆矩阵。from sklearn.metrics import classification_report, confusion_matrix import seaborn as sns import matplotlib.pyplot as plt # y_true, y_pred 来自验证集 print(classification_report(y_true, y_pred, target_names[畅通,缓行,拥堵,严重拥堵])) cm confusion_matrix(y_true, y_pred) sns.heatmap(cm, annotTrue, fmtd, cmapBlues, xticklabels[畅通,缓行,拥堵,严重拥堵], yticklabels[畅通,缓行,拥堵,严重拥堵]) plt.xlabel(预测); plt.ylabel(真实) plt.savefig(confusion_matrix.png, dpi150)逻辑说明classification_report 直接给出每类指标重点看「严重拥堵」的 recall它最低往往说明模型没学到拥堵特征。混淆矩阵看错分方向如果「缓行」大量被分成「拥堵」说明两类视觉边界不清要么合并标签要么加数据。参数上 target_names 顺序必须和 label 编码一致否则图对不上。4.2 学习率与早停的实际设置拥堵数据集规模通常不大学习率用 1e-3 配 AdamW或者 1e-2 配 SGD 加 cosine 退火。早停看验证集 F1 而非 loss因为 loss 降但 F1 不涨说明过拟合。# 以 PyTorch 为例的早停逻辑片段 best_f1 0.0 patience, counter 8, 0 for epoch in range(epochs): train_one_epoch(...) val_f1 evaluate(...) if val_f1 best_f1: best_f1 val_f1 torch.save(model.state_dict(), best_traffic_net.pth) counter 0 else: counter 1 if counter patience: print(fearly stop at epoch {epoch}, best f1{best_f1:.4f}) break逻辑说明保存最佳权重而不是最后一个 epoch这是后悔药。patience 设 8 到 10太小容易早停错过拐点太大浪费算力。参数上监控指标换成 macro-F1 更稳它平等对待每个类别。5. 避坑与排查拥堵识别训练里最常见的 5 个翻车现场5.1 现象训练 loss 正常降验证 accuracy 一直 25% 左右原因类别数没改对或者标签编码和文件夹名映射错位模型在学但评估对不上。解决打印dataset.classes和dataset.class_to_idx确认和模型输出维度一致再检查 DataLoader 的 shuffle 和 drop_last 是否影响验证集。5.2 现象显存溢出batch-size 降到 1 还 OOM原因输入分辨率被源码写死成 512 或更高或者数据增强里有大尺寸随机裁剪。解决在 config 或 transform 里把 Resize 改到 224检查RandomResizedCrop的 scale 参数别让它裁出超大图。另外确认没有在验证阶段保留计算图。5.3 现象严重拥堵类 recall 始终低于 0.3原因类别极度不平衡且增强策略对少数类不友好。解决给少数类单独加增强旋转、亮度扰动、Cutout损失函数加类别权重或者用 WeightedRandomSampler 让少数类每个 epoch 被采到的概率提升。5.4 现象验证集指标很高换一段新路段的视频测试就崩原因过拟合到特定相机视角、光照和道路结构泛化差。解决训练时加随机裁剪、颜色抖动、模糊尽量混入不同时段和天气的数据。如果只有单路段数据别指望模型能跨路段直接用这是拥堵识别的固有边界。5.5 现象训练到一半 loss 变 NaN原因学习率过大、梯度爆炸或者数据里有损坏图片导致异常值。解决加梯度裁剪torch.nn.utils.clip_grad_norm_(model.parameters(), max_norm5.0)学习率降一个量级并在 Dataset 的__getitem__里用 try/except 跳过读不出的图片。6. 把拥堵识别推到可用迁移到自有卡口数据的微调技巧自有数据和开源数据集最大的差别在视角和光照。我一般分两步走先冻结主干只训分类头 5 到 10 个 epoch让新分类头适应你的类别定义再解冻全部用更小的学习率比如 1e-4整体微调。这样比一上来就全量微调稳得多也不容易把预训练特征带偏。# 第一步冻结主干 for name, param in model.named_parameters(): if fc not in name: param.requires_grad False optimizer torch.optim.AdamW(filter(lambda p: p.requires_grad, model.parameters()), lr1e-3) # 训练 5-10 epoch 后解冻 for param in model.parameters(): param.requires_grad True optimizer torch.optim.AdamW(model.parameters(), lr1e-4)逻辑说明冻结阶段只更新分类头收敛快且不会破坏预训练特征。解冻后学习率降 10 倍避免大梯度冲垮主干。参数上冻结 epoch 数看验证 F1 是否平台一般 5 到 10 够。验证方法上别只看离线指标。找一段没参与训练的卡口视频按固定间隔抽帧跑推理把预测等级和人工标注对比重点看等级切换的时间点是否吻合。拥堵是连续变化的模型在等级边界抖动是正常的可以加滑动窗口投票平滑输出。微调策略学习率适用数据量风险只训分类头1e-3每类 100-500 张主干特征不匹配时上限低全量微调1e-4每类 500 张以上小数据易过拟合分层解冻1e-4 到 1e-5中等数据调参成本高最后说个习惯每次改完数据划分或增强策略先跑 3 个 epoch 看验证 F1 有没有异常再决定要不要挂长训练。拥堵识别这活儿数据质量比模型结构重要得多标注边界不清的数据再深的网络也救不回来。希望帮到你。本文还有配套的精品资源点击获取 基于Traffic-Net的交通拥堵程度识别：源码解析与训练避坑指南
简介本资源为基于Traffic-Net训练交通拥堵程度识别的完整项目包面向计算机、人工智能、数据科学等专业的在校学生与教师适合作为课程设计、大作业、毕业设计或入门深度学习实践的参考案例。压缩包共10个文件约6KB以py源码、json模型配置、md说明文档和txt提示文件为主其中Python脚本承载Traffic-Net网络训练与识别逻辑json文件记录模型类别等关键配置md与txt则提供运行说明和常见问题提示结构精简、便于快速理解项目脉络。目前已有50人学习下载。通过该资源读者可掌握交通拥堵程度识别的建模思路与训练流程理解模型配置与数据组织的对应关系并在此基础上进行二次开发或功能扩展积累深度学习项目落地的实践经验。1. 交通拥堵程度识别到底在识别什么从 Traffic-Net 的输入输出说起拿到「基于Traffic-Net训练交通拥堵程度识别源码数据集下载.zip」这个包多数人第一反应是解压、装依赖、跑 train.py。但真正决定你能不能复现出可用结果的不是代码写得多漂亮而是你搞清楚 Traffic-Net 这类模型到底吃什么、吐什么。交通拥堵程度识别本质是一个把道路图像或视频帧映射到拥堵等级的分类/回归任务输入通常是固定视角的卡口或路侧相机画面输出是畅通、缓行、拥堵、严重拥堵这类离散等级或者 0 到 1 之间的拥堵指数。它和车辆检测、车流量统计不是一回事检测关心「有几辆车、在哪」拥堵识别关心「这条路现在堵成什么样」后者对全局纹理、车辆密度、车间距、车道占用率的敏感度远高于单目标框的精度。这套源码加数据集的价值在于它把数据读取、模型定义、训练循环、评估指标打包好了你不需要从零搭一个拥堵分类 pipeline。适合谁做智慧交通课程设计的学生、需要快速验证路侧感知方案的算法工程师、以及手上有自己卡口数据想迁移训练的从业者。但要注意拥堵程度这个标签本身带有主观性不同标注员对「缓行」和「拥堵」的边界判断可能不一致这是你后面调参和看指标时必须心里有数的前提。热搜里常出现的 ccpd数据集、车辆检测数据集bdd100、自动驾驶数据集和拥堵识别是近邻但不同任务别混用标签体系。2. 把 Traffic-Net 的源码和数据集跑起来环境、目录与最小训练命令2.1 解压后先看目录结构别急着 pip install拿到 zip 之后我一般先 tree 一下看目录确认数据集和源码的相对路径。常见结构是Traffic-Net/下分datasets/、models/、train.py、config.yaml、requirements.txt。数据集可能是按类别分文件夹的 ImageFolder 格式也可能是 csv 标注加图片路径。先确认这一点因为后面 DataLoader 怎么写完全取决于它。# 查看压缩包内容不解压先看结构 unzip -l Traffic-Net训练交通拥堵程度识别源码数据集下载.zip | head -50 # 解压到指定目录 unzip Traffic-Net训练交通拥堵程度识别源码数据集下载.zip -d traffic_net_work # 查看目录树两层 cd traffic_net_work find . -maxdepth 2 -type d | sort逻辑说明unzip -l只列出内容不落盘避免解压出一堆乱码文件名后才发现结构不对。参数上-d指定解压目录建议单独建工作目录别在下载目录里直接跑训练否则缓存文件和权重会混在一起。如果压缩包内是中文文件名Linux 下可能乱码用unzip -O GBK或7z x处理。2.2 依赖安装与 CUDA 版本对齐requirements.txt 里通常有 torch、torchvision、numpy、opencv-python、pillow、tqdm、matplotlib。坑在于 torch 版本和 CUDA 驱动不匹配这是血泪经验里排第一的翻车点。# 建虚拟环境别污染系统 python python -m venv venv_traffic source venv_traffic/bin/activate # Windows 用 venv_traffic\Scripts\activate # 先看本机 CUDA 驱动支持的最高版本 nvidia-smi # 按官方对应关系装 torch这里以 cu118 为例按你驱动改 pip install torch torchvision --index-url https://download.pytorch.org/whl/cu118 # 再装其余依赖 pip install -r requirements.txt逻辑说明nvidia-smi右上角 CUDA Version 是驱动支持的上限不是你必须装的版本装低于它的即可。参数上--index-url指向 PyTorch 官方 wheel 源比默认 pypi 快且版本全。如果装完torch.cuda.is_available()返回 False先别怀疑代码九成是 torch 和驱动不匹配重装对应版本即可。2.3 数据集划分与最小训练命令拥堵识别数据集通常按 7:2:1 划分 train/val/test。如果源码没带划分脚本自己按类别分层抽样保证每个拥堵等级在三个集合里比例一致否则验证集指标会虚高。import os, random, shutil from pathlib import Path src Path(datasets/raw) # 原始按类别分好的目录 dst Path(datasets/split) ratios {train: 0.7, val: 0.2, test: 0.1} random.seed(42) for cls_dir in src.iterdir(): if not cls_dir.is_dir(): continue imgs list(cls_dir.glob(*.jpg)) list(cls_dir.glob(*.png)) random.shuffle(imgs) n len(imgs) n_train, n_val int(n * ratios[train]), int(n * ratios[val]) splits { train: imgs[:n_train], val: imgs[n_train:n_train n_val], test: imgs[n_train n_val:], } for split, files in splits.items(): out dst / split / cls_dir.name out.mkdir(parentsTrue, exist_okTrue) for f in files: shutil.copy(f, out / f.name) print(split done)逻辑说明按类别分别抽样再复制保证分层。random.seed(42)固定随机性方便复现。参数上 ratios 可按数据集大小调样本少于 500 张时 test 比例可降到 0.1 以下把更多数据留给训练。复制而非移动保留原始数据后悔药。# 最小训练命令具体参数名以源码 config 为准 python train.py --data datasets/split --epochs 50 --batch-size 32 --lr 1e-3 --img-size 224 --gpu 0逻辑说明--data指向划分后根目录--img-size要和模型输入一致Traffic-Net 常见输入是 224 或 256。--batch-size受显存限制8G 显存跑 224 输入一般 32 到 64。先跑 2 个 epoch 确认 loss 在降、没有 NaN再挂长训练。3. Traffic-Net 模型结构与拥堵等级分类头怎么改3.1 主干网络选型轻量还是精度优先Traffic-Net 这个名字在不同实现里主干可能是 ResNet、MobileNet、EfficientNet 或自研 CNN。路侧设备算力有限时MobileNetV3 或 ShuffleNet 更实际追求精度且用服务器推理ResNet50 起步。判断方法很简单看源码 models 目录里 import 的是哪个 torchvision 模型或者有没有自定义 backbone 文件。主干参数量级适用场景拥堵识别注意点MobileNetV3约 5M边缘设备、实时对纹理细节弱严重拥堵易混ResNet50约 25M服务器、离线精度稳显存占用高EfficientNet-B0约 5M折中方案输入分辨率敏感别乱改自研 CNN看实现特定数据集泛化差迁移需谨慎选型理由拥堵等级区分主要靠车辆密度和排列纹理不需要检测框级别细节所以轻量主干往往够用。但如果你数据集里「缓行」和「拥堵」视觉差异极小就得靠更深的主干或更高输入分辨率。3.2 分类头改造与类别数对齐源码默认类别数可能和你的数据集不一致这是必须改的第一处。假设你的数据集是 4 类拥堵等级而源码写死 3 类不改就报维度错。import torch.nn as nn from torchvision import models num_classes 4 # 畅通/缓行/拥堵/严重拥堵 model models.resnet50(weightsIMAGENET1K_V2) # 替换最后的全连接层 in_features model.fc.in_features model.fc nn.Sequential( nn.Dropout(0.3), nn.Linear(in_features, 256), nn.ReLU(inplaceTrue), nn.Linear(256, num_classes) ) print(model.fc)逻辑说明weightsIMAGENET1K_V2加载预训练权重拥堵数据集通常几千到几万张从头训容易过拟合迁移学习是标配。中间加一层 256 维和 Dropout是为了在小数据集上增加非线性同时抑制过拟合。参数上 Dropout 0.3 到 0.5 之间调数据越少取值越大。类别数一定和数据集文件夹数量一致否则训练不报错但评估全乱。3.3 损失函数与类别不平衡处理拥堵数据天然不平衡畅通样本远多于严重拥堵。直接用 CrossEntropyLoss 会让模型偏向多数类严重拥堵召回率极低。import numpy as np import torch # 统计各类样本数 class_counts np.array([1200, 800, 400, 150]) # 按你的实际统计填 weights 1.0 / class_counts weights weights / weights.sum() * len(class_counts) class_weights torch.tensor(weights, dtypetorch.float32) criterion nn.CrossEntropyLoss(weightclass_weights, label_smoothing0.1)逻辑说明1/类别数再归一化让少数类获得更大权重。label_smoothing0.1缓解标注主观性带来的噪声拥堵等级边界模糊时尤其有用。参数上权重别拉太极端否则少数类过拟合、多数类崩掉一般控制在 10 倍以内。4. 训练过程监控与拥堵识别指标怎么读4.1 该盯哪些曲线不该只看 accuracy拥堵识别里 accuracy 是最容易骗人的指标。四类均衡时 25% 是随机水平但如果畅通占 70%模型全预测畅通也有 70% accuracy实际毫无用处。必须同时看每类的 precision、recall、F1以及混淆矩阵。from sklearn.metrics import classification_report, confusion_matrix import seaborn as sns import matplotlib.pyplot as plt # y_true, y_pred 来自验证集 print(classification_report(y_true, y_pred, target_names[畅通,缓行,拥堵,严重拥堵])) cm confusion_matrix(y_true, y_pred) sns.heatmap(cm, annotTrue, fmtd, cmapBlues, xticklabels[畅通,缓行,拥堵,严重拥堵], yticklabels[畅通,缓行,拥堵,严重拥堵]) plt.xlabel(预测); plt.ylabel(真实) plt.savefig(confusion_matrix.png, dpi150)逻辑说明classification_report 直接给出每类指标重点看「严重拥堵」的 recall它最低往往说明模型没学到拥堵特征。混淆矩阵看错分方向如果「缓行」大量被分成「拥堵」说明两类视觉边界不清要么合并标签要么加数据。参数上 target_names 顺序必须和 label 编码一致否则图对不上。4.2 学习率与早停的实际设置拥堵数据集规模通常不大学习率用 1e-3 配 AdamW或者 1e-2 配 SGD 加 cosine 退火。早停看验证集 F1 而非 loss因为 loss 降但 F1 不涨说明过拟合。# 以 PyTorch 为例的早停逻辑片段 best_f1 0.0 patience, counter 8, 0 for epoch in range(epochs): train_one_epoch(...) val_f1 evaluate(...) if val_f1 best_f1: best_f1 val_f1 torch.save(model.state_dict(), best_traffic_net.pth) counter 0 else: counter 1 if counter patience: print(fearly stop at epoch {epoch}, best f1{best_f1:.4f}) break逻辑说明保存最佳权重而不是最后一个 epoch这是后悔药。patience 设 8 到 10太小容易早停错过拐点太大浪费算力。参数上监控指标换成 macro-F1 更稳它平等对待每个类别。5. 避坑与排查拥堵识别训练里最常见的 5 个翻车现场5.1 现象训练 loss 正常降验证 accuracy 一直 25% 左右原因类别数没改对或者标签编码和文件夹名映射错位模型在学但评估对不上。解决打印dataset.classes和dataset.class_to_idx确认和模型输出维度一致再检查 DataLoader 的 shuffle 和 drop_last 是否影响验证集。5.2 现象显存溢出batch-size 降到 1 还 OOM原因输入分辨率被源码写死成 512 或更高或者数据增强里有大尺寸随机裁剪。解决在 config 或 transform 里把 Resize 改到 224检查RandomResizedCrop的 scale 参数别让它裁出超大图。另外确认没有在验证阶段保留计算图。5.3 现象严重拥堵类 recall 始终低于 0.3原因类别极度不平衡且增强策略对少数类不友好。解决给少数类单独加增强旋转、亮度扰动、Cutout损失函数加类别权重或者用 WeightedRandomSampler 让少数类每个 epoch 被采到的概率提升。5.4 现象验证集指标很高换一段新路段的视频测试就崩原因过拟合到特定相机视角、光照和道路结构泛化差。解决训练时加随机裁剪、颜色抖动、模糊尽量混入不同时段和天气的数据。如果只有单路段数据别指望模型能跨路段直接用这是拥堵识别的固有边界。5.5 现象训练到一半 loss 变 NaN原因学习率过大、梯度爆炸或者数据里有损坏图片导致异常值。解决加梯度裁剪torch.nn.utils.clip_grad_norm_(model.parameters(), max_norm5.0)学习率降一个量级并在 Dataset 的__getitem__里用 try/except 跳过读不出的图片。6. 把拥堵识别推到可用迁移到自有卡口数据的微调技巧自有数据和开源数据集最大的差别在视角和光照。我一般分两步走先冻结主干只训分类头 5 到 10 个 epoch让新分类头适应你的类别定义再解冻全部用更小的学习率比如 1e-4整体微调。这样比一上来就全量微调稳得多也不容易把预训练特征带偏。# 第一步冻结主干 for name, param in model.named_parameters(): if fc not in name: param.requires_grad False optimizer torch.optim.AdamW(filter(lambda p: p.requires_grad, model.parameters()), lr1e-3) # 训练 5-10 epoch 后解冻 for param in model.parameters(): param.requires_grad True optimizer torch.optim.AdamW(model.parameters(), lr1e-4)逻辑说明冻结阶段只更新分类头收敛快且不会破坏预训练特征。解冻后学习率降 10 倍避免大梯度冲垮主干。参数上冻结 epoch 数看验证 F1 是否平台一般 5 到 10 够。验证方法上别只看离线指标。找一段没参与训练的卡口视频按固定间隔抽帧跑推理把预测等级和人工标注对比重点看等级切换的时间点是否吻合。拥堵是连续变化的模型在等级边界抖动是正常的可以加滑动窗口投票平滑输出。微调策略学习率适用数据量风险只训分类头1e-3每类 100-500 张主干特征不匹配时上限低全量微调1e-4每类 500 张以上小数据易过拟合分层解冻1e-4 到 1e-5中等数据调参成本高最后说个习惯每次改完数据划分或增强策略先跑 3 个 epoch 看验证 F1 有没有异常再决定要不要挂长训练。拥堵识别这活儿数据质量比模型结构重要得多标注边界不清的数据再深的网络也救不回来。希望帮到你。本文还有配套的精品资源点击获取