Non-maximum suppression for object detection in a neural network
Switch branches/tags
Nothing to show
Clone or download
Fetching latest commit…
Cannot retrieve the latest commit at this time.
Permalink
Failed to load latest commit information.
.idea
CVPR2017/defense
data
experiments
imdb
nms_net
.gitignore
LICENSE
Makefile
README.md
test.py
train.py

README.md

Learning non-maximum suppression for object detection

This is the code for the paper
Learning non-maximum suppression. Jan Hosang, Rodrigo Benenson, Bernt Schiele. CVPR 2017.

You can find the project page with downloads here: https://mpi-inf.mpg.de/learning-nms

Setup

Run make to compile C++ code and protobufs.

Link to the coco API in the root directory, like so:

/work/src/tf-gnet$ ln -s /work/src/coco/PythonAPI/pycocotools

Link to coco annotations/images in the data subdir:

/work/src/tf-gnet/data$ ln -s /datasets/coco