train_faster-rcnn.sh

  • train_faster-rcnn.sh
CUDA_VISIBLE_DEVICES=${GPU_ID}   time python ./tools/trainval_net.py \
      --weight data/imagenet_weights/${NET}.ckpt \
      --imdb ${TRAIN_IMDB} \
      --imdbval ${TEST_IMDB} \
      --iters ${ITERS} \
      --cfg experiments/cfgs/${NET}.yml \
      --tag ${EXTRA_ARGS_SLUG} \
      --net ${NET} \
      --set ANCHOR_SCALES ${ANCHORS} ANCHOR_RATIOS ${RATIOS} \
      TRAIN.STEPSIZE ${STEPSIZE} ${EXTRA_ARGS}
  • 参数解读
    weight:预训练模型的参数
    imdb:训练的数据集,别名:imdb_name
    imdbval:测试的数据集,别名:imdbval_name
    iters:迭代次数,别名:max_iters
    cfg:网络配置文件,别名:cfg_file
    net:网络模型
    set:设置配置参数——ANCHOR_SCALES和ANCHO_RATIOS,别名:set_cfgs

你可能感兴趣的:(train_faster-rcnn.sh)