Skip to content

A PyTorch implementation of SSD (Liu, Wei, et al. "Ssd: Single shot multibox detector." European conference on computer vision. Springer, Cham, 2016.)

Notifications You must be signed in to change notification settings

kindeQi/PyTorch_SSD

Repository files navigation

PyTorch_SSD

Now it's training on a K80, iter 24k

Current

  • Current mAP: 0.751 (Expected 0.772)
  • Current iteration number: 45k
  • Train from pretrained vgg weight
  • With data augmentation
  • Training dataset: Pascal 07 + 12
  • Val dataset: Pascal 07test
  • Train log: res.json

How to use it

  1. Update config.py, set paths
  2. Command line: python train.py

Reference

Loss Curve

  • Training dataset only
  • Exponentially weighted average

About

A PyTorch implementation of SSD (Liu, Wei, et al. "Ssd: Single shot multibox detector." European conference on computer vision. Springer, Cham, 2016.)

Topics

Resources

Stars

Watchers

Forks

Releases

No releases published

Packages

No packages published