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

你可能感兴趣的:(CenterFusion,python,人工智能,自动驾驶,深度学习,机器学习)