Imported from https://github.com/polarisZhao/PFLD-pytorch