接小咸鱼的文章,http://blog.youkuaiyun.com/sinat_30071459/article/details/51332084:用自己的数据集训练faster rcnn。训练的时候用的./experiments/scripts/faster_rcnn_alt_opt.sh 0 ZF pascal_voc
运行faster_rcnn_alt_opt.sh后会调用tools/train_faster_rcnn_alt_opt.py ,而这个文件中会用到models/passcal_voc/ZF(VGG_16)/faster_rcnn_alt_opt/下面的一串*.pt文件。这些*.pt文件,就是平时caffe用的solver,prototxt文件,即网络配置和网络。
def
parse_args():
主要做的一些提示,GPU id,网络选择、数据选择、weight等。
def
get_roidb(
imdb_name,
rpn_file=
None):
不知道
每个阶段的Solver:
# Solver for each training stage
solvers = [[net_name, n,
'stage1_rpn_solver60k80k.pt'],
[net_name, n,
'stage1_fast_rcnn_solver30k40k.pt'],
[net_name, n,
'stage2_rpn_solver60k80k.pt'],
[net_name, n,
'stage2_fast_rcnn_solver30k40k.pt']]
http://blog.youkuaiyun.com/sloanqin/article/details/51525692