HSAP/algorithms/lane_ufld/code.embedded.bak/UFLD

UFLD 使用说明（lane0_copy 数据）

代码目录：/home/chengfanglu/DATA/BK2/UFLD 数据目录：/home/chengfanglu/DATA/lane0_copy/

一、环境

source ~/miniconda3/etc/profile.d/conda.sh conda activate lane_light cd /home/chengfanglu/DATA/BK2/UFLD

二、数据目录

lane0_copy/ DATASET/（基线包，勿改 list/train_gt.txt） images/ annotations/segmentation_masks/ list/train_gt.txt（训练：图 + mask 两列） list/val_gt.txt list/test_gt.txt list/test.txt（仅图片） DATASET-AddBy-zhangsan-20260615/（增量包，结构同上） lists_merged/（多包合并列表，训练时自动生成） datasets_registry.json（短名别名）

增量包命名：DATASET-AddBy-姓名-YYYYMMDD（日期 8 位，如 20260615） 目录规范详见：/home/chengfanglu/DATA/lane0_copy/DATASETS_LAYOUT.md

新建增量包命令：

python /home/chengfanglu/DATA/lane0_copy/scripts/build_ufld_pack.py --src /path/to/archive --parent /home/chengfanglu/DATA/lane0_copy --engineer zhangsan --date 20260615

别名（config 里可写 DATASET-A）：编辑 lane0_copy/datasets_registry.json，例如： {"aliases": {"DATASET-A": "DATASET-AddBy-zhangsan-20260615"}}

三、配置文件

configs/mufld_lane_multi_pack.py — 推荐，多包训练，用 train_packs 控制合并 configs/mufld_lane_culane.py — 单包，data_root 指向 DATASET 目录本身 configs/mufld_lane_smoke.py — 冒烟（少量样本） configs/tusimple_res18_4lane_v1.py — 对接旧权重 best.pth（griding_num=100）

多包训练请改 configs/mufld_lane_multi_pack.py：

data_root = '/home/chengfanglu/DATA/lane0_copy' train_packs = ['DATASET'] （多包示例：train_packs = ['DATASET', 'DATASET-A']） pack_list_name = 'list/train_gt.txt' remerge_train_list = False（增删包后改为 True，强制重建 lists_merged）

训练时自动合并列表到：lane0_copy/lists_merged/train__DATASET__....txt

四、训练

conda activate lane_light cd /home/chengfanglu/DATA/BK2/UFLD

冒烟： UFLD_NUM_WORKERS=0 python train.py configs/mufld_lane_smoke.py

正式（多包）： python train.py configs/mufld_lane_multi_pack.py

断点续训： python train.py configs/mufld_lane_multi_pack.py --resume log/你的实验目录/best.pth

日志与权重在：log/时间_lr_.../best.pth 无 GPU 时可设 UFLD_NUM_WORKERS=0

常用 config 项： batch_size = 16 learning_rate = 0.1 use_aux = True（False 与旧 best.pth 一致，更省显存） griding_num = 200（旧权重用 100） num_lanes = 4

五、推理与测试

【5.1 可视化 demo】

先准备 test3.txt（示例取 3 张）： awk '{print $1}' /home/chengfanglu/DATA/lane0_copy/DATASET/list/test_gt.txt | head -3 > /home/chengfanglu/DATA/lane0_copy/DATASET/test3.txt

python demo.py configs/tusimple_res18_4lane_v1.py --test_model log/20250702_165153_lr_1e-05_b_32_ufld_2lanes_res18/best.pth --data_root /home/chengfanglu/DATA/lane0_copy/DATASET

【5.2 批量测试】

python test.py configs/tusimple_res18_4lane_v1.py --test_model log/20250702_165153_lr_1e-05_b_32_ufld_2lanes_res18/best.pth --data_root /home/chengfanglu/DATA/lane0_copy/DATASET --test_list list/test_gt.txt

多包时 data_root 用 lane0_copy，例如： python test.py configs/mufld_lane_multi_pack.py --test_model log/xxx/best.pth --data_root /home/chengfanglu/DATA/lane0_copy --test_list lists_merged/train__DATASET.txt

无 test_label.json 时只出预测，不算 TuSimple 官方指标。

【5.3 预测画到图上】

python vis_tusimple_pred.py --pred tmp/tusimple_eval_tmp.0.txt --data_root /home/chengfanglu/DATA/lane0_copy/DATASET --out_dir tmp/vis_pred

六、导出 ONNX

python pth_to_onnx.py --model_path log/20250702_165153_lr_1e-05_b_32_ufld_2lanes_res18/best.pth --output log/20250702_165153_lr_1e-05_b_32_ufld_2lanes_res18/best.onnx

需与训练时 backbone、griding_num、num_lanes 一致。

【6.1 VoVNet backbone】

已从 BK2/archive/vovnet-detectron2-master 移植 OSA+eSE 结构（无 detectron2 依赖），与 ResNet 相同接口。

config backbone	说明
vov19slim	V-19-slim-eSE，约 52.7M 参数（288×800）
vov19slim_dw	slim + depthwise
vov19 / vov39 / vov57 / vov99	更大变体

示例配置：configs/tusimple_vov19slim_4lane_v1.py

python train.py configs/tusimple_vov19slim_4lane_v1.py
python profile_model.py --backbone vov19slim --griding_num 100 --num_lanes 4

VoVNet 无 torchvision 预训练权重，需从头训或自行转换 detectron2 权重。旧 ResNet 的 best.pth 不能直接用于 VoVNet。

七、路径速查

代码：/home/chengfanglu/DATA/BK2/UFLD 数据父目录：/home/chengfanglu/DATA/lane0_copy 基线数据：/home/chengfanglu/DATA/lane0_copy/DATASET 多包配置：configs/mufld_lane_multi_pack.py 已有权重：log/20250702_165153_lr_1e-05_b_32_ufld_2lanes_res18/best.pth