An apparatus for white balance editing, includes a memory storing instructions, and at least one processor configured to execute the instructions to obtain an input image having an original white balance that is corrected by image signal processing, and obtain, using a first neural network, an intermediate representation of the obtained input image, the intermediate representation having the original white balance that is not corrected by the image signal processing. The at least one processor is further configured to execute the instructions to obtain, using a second neural network, a first output image having a first white balance different than the original white balance, based on the obtained intermediate representation.