LSTM-SSD Object Detection in Pytorch

Dependencies

Python 3.6+
OpenCV
Pytorch 1.0 or Pytorch 0.4+

Train

python train_ssd.py --datasets ./data --validation_dataset ./data/ --net vgg16-ssd --batch_size 2 --num_epochs 200 --scheduler cosine --lr 0.01 --t_max 200

Name		Name	Last commit message	Last commit date
Latest commit History 1 Commits
.vscode		.vscode
data		data
vision		vision
.gitignore		.gitignore
LICENSE		LICENSE
README.md		README.md
draw_eval_results.py		draw_eval_results.py
eval_ssd.py		eval_ssd.py
extract_tf_weights.py		extract_tf_weights.py
labels.txt		labels.txt
run_ssd_example.py		run_ssd_example.py
run_ssd_example_output.jpg		run_ssd_example_output.jpg
run_ssd_live_demo.py		run_ssd_live_demo.py
sample.py		sample.py
sample1.py		sample1.py
train_ssd.py		train_ssd.py
visual_tf_models.py		visual_tf_models.py

License

nakajimakou1/lstm-object-detection

Folders and files

Latest commit

History

Repository files navigation

LSTM-SSD Object Detection in Pytorch

Dependencies

Train

About

Resources

License

Stars

Watchers

Forks

Languages