【CenterFusion】训练脚本CenterFusion/experiments/train.sh

• 在 README.md 中训练模型的命令是：bash experiments/train.sh
• 首先执行的就是 train.sh 脚本
• 在脚本中 --参数 值 表示可选参数

cd src
python main.py \
    ddd \
    '''
    可以理解为工作空间名，也就是文件夹名
    '''
    --exp_id centerfusion \
    '''
    项目名称
    '''
    --shuffle_train \
    '''
    将训练模型的数据集进行打乱的操作
    '''
    --train_split train \
    '''
    训练集
    '''
    --val_split mini_val \
    '''
    测试集
    '''
    --val_intervals 1 \
    '''
    运行测试集的 epoch 数为 1
    比如 1000 个样本用来训练一个神经网络，训练就算完成 1 epoch（期）
    '''
    --run_dataset_eval \
    '''
    在 eval 中使用数据集特定的计算函数
    '''
    --nuscenes_att \
    --velocity \
    --batch_size 32 \
    '''
    一次训练抓取 32 个数据样本
    '''
    --lr 2.5e-4 \
    '''
    学习率：0.00025
    '''
    --num_epochs 60 \
    '''
    训练 60 轮
    '''
    --lr_step 50 \
    '''
    学习率步长
    '''
    --save_point 20,40,50 \
    '''
    模型保存时间点
    '''
    --gpus 0,1 \
    '''
    这里使用了两块 GPU ：0 号 GPU 和 1 号 GPU
    '''
    --not_rand_crop \
    '''
    不使用来自 CenterNet 的随机裁剪数据增强
    '''
    --flip 0.5 \
    '''
    使用翻转数据增强的概率 50%
    '''
    --shift 0.1 \
    '''
    当不使用随机裁剪时，10% 概率使用移位增强
    '''
    --pointcloud \
    '''
    雷达点云
    '''
    --radar_sweeps 3 \
    '''
    点云图中雷达扫瞄 3 次
    '''
    --pc_z_offset 0.0 \
    '''
    向 z 轴方面提高雷达点
    '''
    --pillar_dims 1.0,0.2,0.2 \
    '''
    雷达柱尺寸(h、w、l)
    '''
    --max_pc_dist 60.0 \
    '''
    移除最大点云距离 60 以外的点
    '''
    --load_model ../models/centernet_baseline_e170.pth \
    '''
    导入的模型
    '''
cd ..

原文地址：https://blog.csdn.net/qq_34972053/article/details/135703548

免责声明：本站文章内容转载自网络资源，如本站内容侵犯了原著者的合法权益，可联系本站删除。更多内容请关注自学内容网（zxcms.com）！