上一篇：Ubuntu 16.04 測試 tf-faster-rcnn 在CPU下執行

1、在根目錄下開啟終端，輸入以下命令：

（1）如果是使用VGG16模型

mkdir -p data/imagenet_weights
cd data/imagenet_weights
wget -v http://download.tensorflow.org/models/vgg_16_2016_08_28.tar.gz
tar -xzvf vgg_16_2016_08_28.tar.gz
mv vgg_16.ckpt vgg16.ckpt
cd ../..

（2）如果是Resnet101模型

mkdir -p data/imagenet_weights
cd data/imagenet_weights
wget -v http://download.tensorflow.org/models/resnet_v1_101_2016_08_28.tar.gz
tar -xzvf resnet_v1_101_2016_08_28.tar.gz
mv resnet_v1_101.ckpt res101.ckpt
cd ../..
 

2、下載voc2007資料集

http://host.robots.ox.ac.uk/pascal/VOC/voc2007/VOCtrainval_06-Nov-2007.tar
http://host.robots.ox.ac.uk/pascal/VOC/voc2007/VOCtest_06-Nov-2007.tar
http://host.robots.ox.ac.uk/pascal/VOC/voc2007/VOCdevkit_08-Jun-2007.tar

將三個檔案下載解壓到同一個資料夾下：data/VOCdevkit2007/

3、將tools/trainval_net.py拷貝到根目錄下，並修改檔案內容

（1）、註釋：

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

（2）新增default預設引數，省去在Edits Configurations輸入命令引數

 parser.add_argument('--weight', dest='weight', default="./data/imagenet_weights/vgg16.ckpt",
                      help='initialize with pretrained model weights',
                      type=str)

輸出模型路徑可以修改以下引數的deault即可，程式碼預設生成的路徑為：/output/default/voc_2007_trainval/default

parser.add_argument('--tag', dest='tag',
                      help='tag of the model',
                      default=None, type=str)

修改net引數，default="vgg16"

parser.add_argument('--net', dest='net',
                      help='vgg16, res50, res101, res152, mobile',
                      default='vgg16', type=str)

4、相關引數配置檔案config.py

比如引數：__C.TRAIN.SNAPSHOT_ITERS就表示訓練幾次時儲存一次

5、最後開始訓練：