代码地址:
https://github.com.cnpmjs.org/endernewton/tf-faster-rcnn
- 下载代码
git clone https://github.com/endernewton/tf-faster-rcnn.git
- 在安装脚本中更新你的-arch以匹配你的GPU(Update your -arch in setup script to match your GPU)
|GPU model| Architecture |
|-GTX 960m-|-sm_50-|
| GTX 1080 (Ti) | sm_61 |
在lib/set.py 搜索 gpu_nms
-
构建Cython模块(Build the Cython modules)
进入lib文件夹,执行 makeclean make 命令 -
安装Python COCO API。代码需要API来访问COCO数据集。(Install the Python COCO API. The code requires the API to access COCO dataset.)
-
下载VOC数据
https://github.com/rbgirshick/py-faster-rcnn#beyond-the-demo-installation-for-training-and-testing-models
选中它就可以下载