从train_faster_rcnn_alt_opt.py进入。
- 初始化参数:args = parse_args() 采用的是Python的argparse
主要有–net_name、–gpu、–cfg等(在cfg中只是修改了几个参数,其他大部分参数默认在congig.py中,涉及到训练整个网络)。 - cfg_from_file(args.cfg_file) 这里便是代用config中的函数cfg_from_file来读取前面cfg文件中的参数,同时调用_merge_a_into_b函数把所有的参数整合,其中__C = edict() cfg = __C cfg是一个词典(edict)数据结构。
- faster rcnn采用的是多进程,mp_queue是进程间用于通讯的数据结构
- 1
- 2
同时solvers, max_iters, rpn_test_prototxt = get_solvers(args.net_name)得到solver参数
接下来便正式进入了训练的各个阶段。