Abstract: A method for performing multi-functional image restoration by an electronic device with a trained Machine Learning (ML) model is provided. The method includes receiving an image and determining channels of the image, and determining whether a number of restructuring needed for the channels is one. When the restructuring needed for the channels not one, then the method includes restructuring each channel into a first channel set, generating first inferences of the image corresponding to each channel by feeding the first channel set to the trained ML model, and generating a final inference image by combining the first inferences. When the number of restructuring needed for the channels is one, then the method includes restructuring the channels into a second channel set, and generating a second inference of the image by feeding the second channel set to the trained ML model.