TensorFlow_Faster_RCNN中demo.py的运行(CPU Only) 2. 修改代码为CPU Only 3. 在./lib目录下建立Cython模块 4. 在 ./data 下安装Python COCO API 5. 下载数据 6. 下载预训练模型 7. 建立预训练模型的软链接 8. 运行./tools/demo.py
GitHub项目地址,https://github.com/endernewton/tf-faster-rcnn
Tensorflow Faster RCNN for Object Detection
.
本机环境:
- Mac: 10.13.4 无GPU
- python: 3.5
- tensorflow: 1.11.0
1 |
git clone https://github.com/endernewton/tf-faster-rcnn.git |
①修改./lib/setup.py
注释掉第55、87、120-136行与cuda、GPU相关的代码
②./lib/model/nms_wrapper.py
注释掉第12、20-21行
3. 在./lib目录下建立Cython模块
清除上一次编译产生的可执行文件(.pyc、.so)
1 |
make clean |
若不报错则编译成功。会生成.so文件、“pycache”下的.pyc文件
1 |
make |
返回上一级
1 |
cd .. |
4. 在 ./data 下安装Python COCO API
1 |
git clone https://github.com/pdollar/coco.git |
5. 下载数据
setup VOC and COCO datasets (Part of COCO is done)
1 |
cd ./data |
完成以上步骤后,data文件夹下新增2个文件夹:
- VOCdevkit(包含文件夹VOCcode、VOC2007及一些其他文件)
- VOCdevkit2007 (软链接,点击后跳转至./data/VOCdevkit/)
6. 下载预训练模型
./data/scripts/下的.sh文件中的URL无法访问,可以通过作者提供的Google Drive下载。
选择res101中的voc_0712_80k-110k.tgz,下载到./data中。
7. 建立预训练模型的软链接
1 |
NET=res101 |
这6行代码的功能为:
①建立文件夹NewDir = ./output/res101/voc_2007_trainval+voc_2012_trainval/
②在新建文件夹的 ${NewDir}/default 中链接下载的预训练模型,即./data/voc_2007_trainval+voc_2012_trainval 中的4个参数文件
这时,点击./output/vgg16/coco_2014_train+coco_2014_valminusminival/default可以跳转至./data/coco_2014_train+coco_2014_valminusminival 的4个参数文件
8. 运行./tools/demo.py
对./data/demo下的图片进行检测
得到图片的主要对象及置信概率: