代码地址
VGG16参数
1、安装tensorflow,最好是GPU版本。按照说明进行操作。如果未安装GPU版本,则需要注释掉代码中的所有GPU调用,并用相对的CPU调用替换它们
2、Checkout this branch
3、Install python packages (cython, python-opencv, easydict) by running
安装Cython: pip install Cython
安装cv2: pip install opencv-python
安装easydict: pip install easydict
4、Go to ./data/coco/PythonAPI,在终端里运行下面的语句:
python setup.py build_ext --inplace
python setup.py build_ext install
Go to ./lib/utils,在终端里运行下面的语句:
python setup.py build_ext --inplace
5、数据集
(1)下载训练、验证、测试的数据集
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
(2)将所有这些tar解压缩到一个目录中 VOCdevkit,按照下面的结构存放:
data\VOCDevkit2007\VOC2007
6、下载经过训练的VGG16参数地址 ,并将其按下面结构放置:data\imagenet_weights\vgg16.ckpt。
7、运行train.py