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

  • ?p>

    ?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 ..