Excellent work! Does the author need to train the image restoration model by himself? Is there a trained one that can be tested directly?