Fast RCNN训练阶段代码解析

最新推荐文章于 2018-01-09 20:43:02 发布

陈龙CL

最新推荐文章于 2018-01-09 20:43:02 发布

阅读量3k

点赞数 3

分类专栏： caffe python

本文链接：https://blog.csdn.net/timeflyhigh/article/details/51790718

版权

本文介绍了Fast RCNN训练过程，从入口文件train_net.py开始，详细讲解了数据的读取与处理，包括数据集的制作和网络训练。在lib目录下，数据处理位于lib/datasets和lib/fast_rcnn中，而网络输入层由lib/roi_data_layer的python实现完成。训练数据通过get_imdb函数读取，然后使用get_training_roidb进行预处理。同时，文章提到了fast-rcnn如何在Caffe中使用Python实现网络层。

摘要由CSDN通过智能技术生成

首先是入口文件trian_net.py，真正处理数据的文件都在lib文件里，包括数据集制作的文件在lib/datasets下，网络训练测试的文件在lib/fast_rcnn下，lib/roi_data_layer是用python实现的网络的输入层。

parse_args函数解析输入参数：网络参数定义，初始化模型（这两项没有默认值必须自己指定），显卡号，最大迭代次数，训练数据位置等。
def parse_args():
    """
    Parse input arguments
    """
    parser = argparse.ArgumentParser(description='Train a Fast R-CNN network')
    parser.add_argument('--gpu', dest='gpu_id',
                        help='GPU device id to use [0]',
                        default=0, type=int)
    parser.add_argument('--solver', dest='solver',
                        help='solver prototxt',
                        default=None, type=str)
    parser.add_argument('--iters', dest='max_iters',
                        help='number of iterations to train',
                        default=40000, type=int)
    parser.add_argument('--weights', dest='pretrained_model',
                        help='initialize with pretrained model weights',
                        default=None, type=str)
    parser.add_argument('--cfg', dest='cfg_file',
                        help='optional config file',
                        default=None, type=str)
    parser.add_argument('--imdb', dest='imdb_name',
                        help='dataset to train on',
                        default='voc_2007_trainval', type=str)
    parser.add_argument('--rand', dest='randomize',
                        help='randomize (do not use a fixed seed)',
                        action='store_true')
    parser.add_argument