SSD pytorch https://github.com/amdegroot/ssd.pytorch