Imported from https://github.com/Ontheway361/pfld-pytorch