Fixed version of https://github.com/tomguluson92/PRNet_PyTorch - View it on GitHub