Windows10+Faster-RCNN-TensorFlow-Python3-master+VOC2007数据集

Posted 2021-12-22 大彤小忆

• 一、所需文件下载链接
• 二、基础环境配置
• 三、训练及测试过程

  使用Faster R-CNN算法在VOC2007数据集上实现目标检测。

一、所需文件下载链接

1. Faster R-CNN源码及操作步骤Github链接→Faster-RCNN-TensorFlow-Python3。
2. Faster-RCNN-TensorFlow-Python3-master压缩包百度云盘链接→提取码：76wq。
3. VOC2007数据集百度云盘链接→提取码：z8sd。
   VOC2007数据集的解析→VOC2007数据集详细分析。

二、基础环境配置

1. Windows10 + Anaconda3 + PyCharm 2019.3.3
2. 安装CPU版本的TensorFlow
3. 在PyCharm中配置好TensorFlow环境

三、训练及测试过程

1. 在配置好TensorFlow环境的PyCharm中打开下载并解压好的Faster-RCNN-TensorFlow-Python3-master项目。
    
   

2. 安装源码运行需要的Python包。打开j好的Faster-RCNN-TensorFlow-Python3-master项目中有一个requirement.txt文件，其中记录了需要安装的包的名字。在PyCharm的终端Terminal中输入pip install -r requirements.txt后回车，安装需要的所有依赖包。
    
   

3. 在PyCharm的终端Terminal中输入cd data\\coco\\PythonAPI切换路径到./data/coco/PythonAPI，然后输入python setup.py build_ext --inplace并回车。
    
    
   再输入python setup.py build_ext install并回车。
    
   

4. 在PyCharm的终端Terminal中输入exit退出当前路径，然后输入cd lib\\utils切换路径到./lib/utils，再输入python setup.py build_ext --inplace并回车。
    
   

5. 将解压好的VOC2007数据集文件夹VOCDevkit重命名为VOCDevkit2007，并复制到./data文件夹下。
    
   

6. 下载预训练模型VGG16，在./data文件夹下新建文件夹imagenet_weights，将下载好的vgg_16_2016_08_28.tar.gz解压到./data/imagenet_weights路径下，并将vgg_16.ckpt重命名为vgg16.ckpt。
    
   

7. 修改config.py文件。在./lib/config文件夹下的config.py文件，是专门的配置文件，其中定义了模型的诸多参数，大家可以根据自己的需要修改相关参数。在这里，为了减少训练时间，我将最大迭代次数max_iters参数由40000改为10000，同时将迭代多少次保存一次模型snap_iterations参数由5000改为2000，其他参数未作改变。
    
   

8. 在PyCharm中打开train.py文件 ，如果后面想要可视化loss的话，需要在如下图所示的两处添加代码，如果不需要的话，可忽略这一步。
    
   

9. 在train.py文件界面，右击后点击Run 'train'，开始训练。需要注意的是，每次训练前都要清空./data/cache和./default/voc_2007_trainval/default文件夹里面的文件。
    
   

10. 训练结束后，如果训练前在train.py文件中加了可视化loss的代码的话，可以在根路径下得到一个write_loss.txt，保存着每迭代10次对应的损失值。在根路径下新建一个Python文件visual_loss.py，代码如下，运行后可以在根目录下得到一张loss曲线图。

import numpy as np
import matplotlib.pyplot as plt

y_ticks = [0, 0, 0.5, 1.0, 2.0, 3.0, 4.0, 5.0]  # 纵坐标的值，可以自己设置。
data_path = 'E:\\\\Remote Sensing\\\\Faster-RCNN-TensorFlow-Python3-master\\\\write_loss.txt'  # log_loss的路径
result_path = 'E:\\\\Remote Sensing\\\\Faster-RCNN-TensorFlow-Python3-master\\\\total_loss'  # 保存结果的路径

data1_loss = np.loadtxt(data_path)
x = data1_loss[:, 0]  # 冒号左边是行范围，冒号右边列范围，取第一列
y = data1_loss[:, 1]  # 取第2列

# 开始画图
fig = plt.figure()
ax = fig.add_subplot(1, 1, 1)
ax.plot(x, y, label='total_loss')
plt.yticks(y_ticks)  # 如果不想自己设置纵坐标，可以注释掉
ax.legend(loc='best')
ax.set_title('The loss curves')
ax.set_xlabel('batches')
fig.savefig(result_path)

11. 训练时，模型保存的路径是./default/voc_2007_trainval/default，每次保存模型都是保存4个文件。
     
    

12. 新建./output/vgg16/voc_2007_trainval/default文件夹，从./default/voc_2007_trainval/default路径下复制一组模型数据到新建的文件夹下，并将所有文件名改为vgg16.后缀。
     
    

13. 准备开始测试已经训练好的模型。在PyCharm中打开demo.py文件，并进行如下修改。
      
    运行demo.py，可以得到./data/demo文件夹下的样例图片的目标检测结果。
     

14. 上述得到的结果中，对于同一幅图像，每个类别会显示一个窗口，因此会显示很多个图片窗口。如果想将同一张图片上的所有目标检测框全部在一张图片上显示出来的话，需要再次修改demo.py文件如下所示的部分。
     
     
    再次运行demo.py，可以重新得到./data/demo文件夹下的样例图片的目标检测结果。可以发现，所有画出来的候选框均在一幅图中标注出来。
     
    

15. 如果想要在测试时输出PR曲线并计算AP值的话，需要在以下几个文件中做如下修改，并新建一个test_net.py文件。
    首先在.\\lib\\datasets路径下打开pascal_voc.py文件，做如下修改。
     
     
    接下来在.\\lib\\datasets路径下打开voc_eval.py文件，做如下修改。
     
     
    然后在根路径下新建test_net.py文件，代码如下所示。

""""
Demo script showing detections in sample images.
See README.md for installation instructions before running.
"""
from __future__ import absolute_import
from __future__ import division
from __future__ import print_function

import argparse
import os

import tensorflow as tf
from lib.nets.vgg16 import vgg16
from lib.datasets.factory import get_imdb
from lib.utils.test import test_net

NETS = 'vgg16': ('vgg16.ckpt',)  # 自己需要修改：训练输出模型
DATASETS = 'pascal_voc': ('voc_2007_trainval',), 'pascal_voc_0712': ('voc_2007_trainval+voc_2012_trainval',)


def parse_args():
    """Parse input arguments."""
    parser = argparse.ArgumentParser(description='Tensorflow Faster R-CNN test')
    parser.add_argument('--net', dest='demo_net', help='Network to use [vgg16 res101]',
                        choices=NETS.keys(), default='vgg16')
    parser.add_argument('--dataset', dest='dataset', help='Trained dataset [pascal_voc pascal_voc_0712]',
                        choices=DATASETS.keys(), default='pascal_voc')
    args = parser.parse_args()
    return args


if __name__ == '__main__':
    args = parse_args()
    # model path
    demonet = args.demo_net
    dataset = args.dataset
    tfmodel = os.path.join('output', demonet, DATASETS[dataset][0], 'default', NETS[demonet][0])  # 模型路径
    # 获得模型文件名称
    filename = (os.path.splitext(tfmodel)[0]).split('\\\\')[-1]
    filename = 'default' + '/' + filename
    imdb = get_imdb("voc_2007_test")  # 得到
    imdb.competition_mode('competition mode')
    if not os.path.isfile(tfmodel + '.meta'):
        print(tfmodel)
        raise IOError((':s not found.\\nDid you download the proper networks from '
                       'our server and place them properly?').format(tfmodel + '.meta'))
    # set config
    tfconfig = tf.ConfigProto(allow_soft_placement=True)
    tfconfig.gpu_options.allow_growth = True
    # init session
    sess = tf.Session(config=tfconfig)
    # load network
    if demonet == 'vgg16':
        net = vgg16(batch_size=1)
    # elif demonet == 'res101':
    #     net = resnetv1(batch_size=1, num_layers=101)
    else:
        raise NotImplementedError
    net.create_architecture(sess, "TEST", 21,  # 自己需要修改：类别数量+1
                            tag='default', anchor_scales=[8, 16, 32])
    saver = tf.train.Saver()
    saver.restore(sess, tfmodel)
    print('Loaded network :s'.format(tfmodel))
    print(filename)
    test_net(sess, net, imdb, filename, max_per_image=100)
    sess.close()

16. 在test_net.py文件界面，右击后点击Run 'test_net'，开始测试。需要注意的是，每次测试前都要清空./data/VOCdevkit2007/annotation_cache文件夹里面的文件。
     
     
    最后可以得到每个类别各自的AP及mAP值（由于迭代的次数比较少，所以结果不是很好）。
     
     
    及每个类别对应的PR曲线（由于图例设置的原因，导致输出的PR曲线如下图所示，可以自行更改）。
     
    

  整个Faster R-CNN模型在VOC2007数据集上的训练与测试到此就结束了。

  参考文章：https://blog.csdn.net/water_93/article/details/106364209
       https://blog.csdn.net/weixin_45798949/article/details/105794168
       https://blog.csdn.net/10km/article/details/68926498