Skip to content

CharlesShang/ReInspect

 
 

Repository files navigation

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 or the video for a demonstration.

Installation & Demo

$ # install the apollocaffe pull request on caffe. see apollocaffe.com
$ git clone http://github.com/russell91/reinspect
$ cd reinspect
$ make eval

Running make eval will download the data and start a notebook to visualize the performance of the model. If you want to train the model from scratch, you can use

$ make train

Running on your own data

The easiest way to run on your own data is to resize your images to 480x640 and provide labels for each object in each image with the idl text files.

Once you have verified that you can train on 480x640 images, you can also modify the image width and height options in config.json. We recommend you choose an image size which is an integer multiple of 32, and then modify the (15, 20) grid to (image_height / 32, image_width / 32).

About

End to end detection in crowded scenes

Resources

Stars

Watchers

Forks

Releases

No releases published

Packages

No packages published

Languages

  • Jupyter Notebook 95.5%
  • Python 4.0%
  • Other 0.5%