Imported from https://github.com/shivamsaboo17/Deep-Restore-PyTorch