Imported from https://github.com/kaichengyan/facenet-darknet-pytorch