SSD

Minimal SSD implemented with MXNet/Gluon and nvidia DALI.

train

python train.py --cfg ./cfgs/ssd_mobilenet_1.0_coco_512x512.yaml

backbone	input_size	mAP(0.5:0.95)
mobilenet1.0	512	24.2

more models will be trained if I have enough GPU ...

Name		Name	Last commit message	Last commit date
Latest commit History 12 Commits
cfgs		cfgs
output		output
samples		samples
.gitignore		.gitignore
README.md		README.md
anchor.py		anchor.py
coco_metric.py		coco_metric.py
demo.ipynb		demo.ipynb
detection.py		detection.py
export_helper.py		export_helper.py
feature.py		feature.py
logger.py		logger.py
loss.py		loss.py
mscoco.py		mscoco.py
net.py		net.py
solver.py		solver.py
target.py		target.py
train.py		train.py