py-faster-rcnn流程(1)——准备阶段

来源:互联网 发布:琢磨先生妻子淘宝店 编辑:程序博客网 时间:2024/05/22 00:28

    从train_faster_rcnn_alt_opt.py进入。

  1.     初始化参数:args = parse_args() 采用的是Python的argparse 
        主要有–net_name、–gpu、–cfg等(在cfg中只是修改了几个参数,其他大部分参数默认在congig.py中,涉及到训练整个网络)。
  2.      cfg_from_file(args.cfg_file) 这里便是代用config中的函数cfg_from_file来读取前面cfg文件中的参数,同时调用_merge_a_into_b函数把所有的参数整合,其中__C = edict() cfg = __C cfg是一个词典(edict)数据结构。
  3.     faster rcnn采用的是多进程,mp_queue是进程间用于通讯的数据结构
import multiprocessing as mpmp_queue = mp.Queue()
  • 1
  • 2

    同时solvers, max_iters, rpn_test_prototxt = get_solvers(args.net_name)得到solver参数 
   接下来便正式进入了训练的各个阶段。

阅读全文
0 0
原创粉丝点击