Method and device for correcting lighting of image

An image processing device comprising:

• a communicator configured to receive an input image;
• and
• a processor configured to
• input the input image to a first neural network and generate predicted lighting data corresponding to lighting of the input image and embedding data corresponding to a feature of the input image, wherein the predicted lighting data is represented as a three-dimensional (3D) vector, with three components representing red color value, green color value, and blue color value of the lighting applied to content of the input image,
• input the generated predicted lighting data, the generated embedding data, and sensor data to a second neural network and generate a lighting weight corresponding to the input image, wherein the sensor data indicates an amount of light measured by a sensor, wherein the lighting weight is represented as a single vector including a plurality of lighting weight values, and
• generate correction lighting data for the input image in response to applying the generated lighting weight to preset basis lighting data corresponding to the input image, wherein the preset basis lighting data is determined based on a color temperature of light emitted from a light source or a type of a capturing device that captured the input image, wherein the preset basis lighting data includes a plurality of vectors, and a number of vectors included in the basis lighting data matches the number of lighting weight values of the lighting weight vector,
• wherein the processor is further configured to:
• determine a first lighting correction vector by calculating a weighted sum between the plurality of vectors of the preset basis lighting data and the plurality of lighting weight values of the lighting weight;
• and
• generate the correction lighting data for the input image based on the first lighting correction vector.