Faster RCNN 代码详解——tf_faster_RCNN（一）

论文地址

link

代码地址

link

视频讲解

link

代码包结构介绍

• data （存放训练和测试数据）
• docker （存放docker文件，配置环境用）
• experiments （存放些参数文件、日志、实验脚本）
• lib （库文件）
• tools （训练测试评估脚本）

tools文件夹下的trainval_net.py文件的main函数

首先从tools文件夹下的trainval_net.py文件的main函数开始读代码

if __name__ == '__main__':
    args = parse_args()                                        #---参数解析

    print('Called with args:')
    print(args)                                                #---显示参数

parse_args()函数

def parse_args():
  """
  Parse input arguments
  """
  parser = argparse.ArgumentParser(description='Train a Fast R-CNN network')
  parser.add_argument('--cfg', dest='cfg_file',
                      help='optional config file',
                      default=None, type=str)
  parser.add_argument('--weight', dest='weight',
                      help='initialize with pretrained model weights',
                      type=str)
  parser.add_argument('--imdb', dest='imdb_name',
                      help='dataset to train on',
                      default='voc_2007_trainval', type=str)
  parser.add_argument('--imdbval', dest='imdbval_name',
                      help='dataset to validate on',
                      default='voc_2007_test', type=str)
  parser.add_argument('--iters', dest='max_iters',
                      help='number of iterations to train',
                      default=70000, type=int)
  parser.add_argument('--tag', dest='tag',
                      help='tag of the model',
                      default=None, type=str)
  parser.add_argument('--net', dest='net',
                      help='vgg16, res50, res101, res152, mobile',
                      default='res50', type=str)
  parser.add_argument('--set', dest='set_cfgs',
                      help='set config keys', default=None,
                      nargs=argparse.REMAINDER)

  if len(sys.argv) == 1:
    parser.print_help()
    sys.exit(1)

  args = parser.<