Skip to content

pranoyr/lstm-object-detection

Repository files navigation

LSTM-SSD Object Detection in Pytorch

Dependencies

  1. Python 3.6+
  2. OpenCV
  3. Pytorch 1.0 or Pytorch 0.4+

Train

python train_ssd.py --datasets /home/neuroplex/data/ILSVRC/ --validation_dataset ./data/ --net lstm-mobilenet --batch_size 1 --num_epochs 200 --gpu 1

About

Object Detection using LSTM-SSD

Resources

License

Stars

Watchers

Forks

Releases

No releases published

Packages

No packages published

Languages