上一篇: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、最后开始训练: