GitXplorerGitXplorer
f

RefineDet_chainer

public
14 stars
6 forks
0 issues

Commits

List of commits on branch master.
Unverified
894711f8bfef6702d208cdf2e758b5fc50c74e79

bug fix

ffukatani committed 6 years ago
Unverified
11a13021a32aeb603393057d8fe25e98926c78b2

bug fix

ffukatani committed 7 years ago
Unverified
9c84bf9d61b6ab9c8ef349ce34f9c59a0dcd14f3

bug

ffukatani committed 7 years ago
Unverified
daf784a10a4a7b695d5ee9074b29f8a3f2895ed1

fix two cascaded regression

ffukatani committed 7 years ago
Unverified
106c0a3603ee64504bd7db111e3868ed8ec14d87

gitignore

ffukatani committed 7 years ago
Unverified
ae03b3e9247df353257bd8c7cdeb6e818cf46e97

bug fix

ffukatani committed 7 years ago

README

The README file for this repository.

RefineDet_chainer

Benchmarking RefineDet[1] and other SSD (Single shot Detection) Network based on chainer

Including

  • DSSD[2]
  • SSD with residual prediction module[2]
  • ESSD[3]
  • RefineDet[1]

Original DSSD is based on ResNet 101. Since memory limitation, only tried on VGG.

Benchmark result

Model name Base model Input image size mAP mAP(paper)
SSD VGG16 300x300 77.5 77.5
SSD Plus(Use Residual Prediction module) VGG16 300x300 78.0 NA
ESSD VGG16 300x300 78.8 79.4
RefineDet VGG16 320x320 Now evaluating 80.0

*: I set batchsize to 22 because of memory limitation. The original paper used 32.

*: Some training condition is different from paper.

*: ESSD original paper did 3 stages training (Only SSD, Only extensional module and whole network), but I did whole training only.

*: I may mistook unintensionally.

Requirement

Usage

git clone https://github.com/fukatani/RefineDet_chainer
cd refinedet
python train.py --model refinedet320 --batchsize 22

Other

Many implementation is referenced chainercv. Thanks!

Reference