Imported from https://github.com/ronghuaiyang/arcface-pytorch