darkseed / ReInspect

Geek Repo:Geek Repo

Github PK Tool:Github PK Tool

ReInspect

ReInspect is an neural network extension to Overfeat-GoogLeNet in Caffe. It is designed for high performance object detection in images with heavily overlapping instances. See the paper for details.

Installation

ReInspect depends on Caffe and requires the ApolloCaffe pull request.

$ git clone http://github.com/russell91/apollocaffe
$ cp <your caffe Makefile.config> apollocaffe/Makefile.config
$ cd apollocaffe && make -j8
$ export PYTHONPATH=/path/to/apollocaffe/python:$PYTHONPATH
$ export LD_LIBRARY_PATH=/path/to/apollocaffe/build/lib:$LD_LIBRARY_PATH

With ApolloCaffe installed, you can run ReInspect with:

$ git clone http://github.com/russell91/reinspect
$ cd reinspect
$ python train.py --config config.json --gpu -1

Data should be placed in /path/to/reinspect/data/ and can be found here.

Evaluation

To evaluate ReInspect, we provide an ipython notebook to visualize the performance of the model.

About


Languages

Language:Python 99.3%Language:Shell 0.7%