当前位置：首页 > news >正文

快速上手：在星图AI上训练PETRV2-BEV模型，实现3D目标检测

news 2026/4/29 7:18:00

快速上手：在星图AI上训练PETRV2-BEV模型，实现3D目标检测

1. 环境准备与快速部署

1.1 激活预置环境

星图AI平台已为我们准备好完整的训练环境，只需简单激活即可使用：

conda activate paddle3d_env

验证环境是否正常：

python -c "import paddle; print(paddle.__version__)" # 应输出类似：2.5.0

1.2 进入工作目录

所有操作将在Paddle3D项目目录下进行：

cd /usr/local/Paddle3D

2. 数据与模型准备

2.1 下载预训练权重

获取官方提供的PETRv2预训练模型：

wget -O /root/workspace/model.pdparams https://paddle3d.bj.bcebos.com/models/petr/petrv2_vovnet_gridmask_p4_800x320/model.pdparams

2.2 获取nuScenes mini数据集

为快速验证流程，我们先使用小型数据集：

wget -O /root/workspace/v1.0-mini.tgz https://www.nuscenes.org/data/v1.0-mini.tgz mkdir -p /root/workspace/nuscenes tar -xf /root/workspace/v1.0-mini.tgz -C /root/workspace/nuscenes

3. 训练流程实战

3.1 数据预处理

生成模型所需的标注缓存文件：

rm /root/workspace/nuscenes/petr_nuscenes_annotation_* -f python3 tools/create_petr_nus_infos.py \ --dataset_root /root/workspace/nuscenes/ \ --save_dir /root/workspace/nuscenes/ \ --mode mini_val

3.2 初始精度评估

测试预训练模型在mini集上的表现：

python tools/evaluate.py \ --config configs/petr/petrv2_vovnet_gridmask_p4_800x320_nuscene.yml \ --model /root/workspace/model.pdparams \ --dataset_root /root/workspace/nuscenes/

3.3 启动模型训练

执行微调训练命令：

python tools/train.py \ --config configs/petr/petrv2_vovnet_gridmask_p4_800x320_nuscene.yml \ --model /root/workspace/model.pdparams \ --dataset_root /root/workspace/nuscenes/ \ --epochs 100 \ --batch_size 2 \ --log_interval 10 \ --learning_rate 1e-4 \ --save_interval 5 \ --do_eval

3.4 训练监控

启动可视化工具查看训练曲线：

visualdl --logdir ./output/ --host 0.0.0.0

通过SSH端口转发访问：

ssh -p 31264 -L 0.0.0.0:8888:localhost:8040 root@gpu-09rxs0pcu2.ssh.gpu.csdn.net

4. 模型导出与验证

4.1 导出推理模型

将训练好的模型转换为部署格式：

rm -rf /root/workspace/nuscenes_release_model mkdir -p /root/workspace/nuscenes_release_model python tools/export.py \ --config configs/petr/petrv2_vovnet_gridmask_p4_800x320_nuscene.yml \ --model output/best_model/model.pdparams \ --save_dir /root/workspace/nuscenes_release_model

4.2 运行可视化DEMO

查看模型实际检测效果：

python tools/demo.py /root/workspace/nuscenes/ /root/workspace/nuscenes_release_model nuscenes

5. 进阶训练（可选）

5.1 XTREME1数据集训练

对于更复杂的城市场景：

python tools/train.py \ --config configs/petr/petrv2_vovnet_gridmask_p4_800x320.yml \ --model /root/workspace/model.pdparams \ --dataset_root /root/workspace/xtreme1_nuscenes_data/ \ --epochs 100 \ --batch_size 2 \ --log_interval 10 \ --learning_rate 1e-4 \ --save_interval 5 \ --do_eval