Awesome
tf-faster-rcnn
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.
Currently, this repository supports Python 3.5, 3.6 and 3.7. Thanks to @morpheusthewhite
PLEASE BE AWARE: I don't have the time or intention to fix all the issues for this branch because I don't use it commercially. I created this branch just for fun. If you want to make any commitment, it is more than welcome. Tensorflow has already released an object detection API. Please refer to it. https://github.com/tensorflow/models/tree/master/research/object_detection
How To Use This Branch
-
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 asconda
you should follow its instruction) -
Go to ./data/coco/PythonAPI
Runpython setup.py build_ext --inplace
Runpython setup.py build_ext install
Go to ./lib/utils and runpython 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.