This repo implements PyramidBox with pytorch
Branch: master
Clone or download
Permalink
Type Name Latest commit message Commit time
Failed to load latest commit information.
data init Jul 25, 2018
layers init Jul 25, 2018
utils init Jul 25, 2018
README.md Update README.md Jul 30, 2018
pyramid.py init Jul 25, 2018
test.py init Jul 25, 2018
train.py fix bug Aug 28, 2018

README.md

PyramidBox

This repo implements PyramidBox with pytorch The paper PyramidBox is based on VGG16. But this repo is based on resnet50.

Here, I implements data-anchor-sampling, LFPN, max-in-out layer.

The result on WIDER FACE Val set:

AP Easy AP Medium AP Hard
95.3 94.3 89.0

My model is released now!

You can download model from Baidu!

With the model and test.py, you can get the same result on WIDER FACE Val set!

Usage

Prerequisites

*Python3

*Pytorch3.1

*OpenCV3

annoPath

annoPath is path to your label file.

The label file should in the following format:

path_to_img1 num_face1 X1 Y1 W1 H1 X2 Y2 ... Wn1 Hn1

path_to_img2 num_face2 X1 Y1 W1 H1 X2 Y2 ... Wn2 Hn2