Tensorflow Faster R-CNN for Windows and Linux by using Python 3
This is the branch to compile Faster R-CNN on Windows and Linux. It is heavily inspired by the great work done here and here. I have not implemented anything new but I fixed the implementations for Windows, Linux and Python 3.
Currently, this repository supports Python 3.5, 3.6 and 3.7. Thanks to @morpheusthewhite
Install tensorflow, preferably GPU version. Follow instructions. If you do not install GPU version, you need to comment out all the GPU calls inside code and replace them with relavent CPU ones.
Checkout this branch
Install python packages (cython, python-opencv, easydict) by running
pip install -r requirements.txt
(if you are using an environment manager system such as conda
you should follow its instruction)
Go to ./data/coco/PythonAPI
Run python setup.py build_ext --inplace
Run python setup.py build_ext install
Go to ./lib/utils and run python setup.py build_ext --inplace
Follow these instructions to download PyCoco database.
I will be glad if you can contribute with a batch script to automatically download and fetch. The final structure has to look like
data\VOCDevkit2007\VOC2007
Download pre-trained VGG16 from here and place it as data\imagenet_weights\vgg16.ckpt
.
For rest of the models, please check here
Run train.py
Notify me if there is any issue found.
此处可能存在不合适展示的内容,页面不予展示。您可通过相关编辑功能自查并修改。
如您确认内容无涉及 不当用语 / 纯广告导流 / 暴力 / 低俗色情 / 侵权 / 盗版 / 虚假 / 无价值内容或违法国家有关法律法规的内容,可点击提交进行申诉,我们将尽快为您处理。