Multi-plane augmented reality image generation

A system, comprising:

• a spatial light modulator (SLM) arranged to output, onto a vehicle windshield, an augmented reality (AR) image including a plurality of sub-images each output in one of a plurality of focal planes;
• an image sensor positioned to obtain a feedback image of the AR image;
• and
• a computer including a processor and a memory, the memory storing instructions executable by the processor programmed to:
• input a set of training images into a neural network that outputs a pixel-wise phase matrix identifying pixels in the training images, wherein each training image corresponds to one respective sub-image;
• actuate the SLM to output the AR image based on the pixel-wise phase matrix;
• determine an offset based on comparing the training images to the feedback image;
• and
• update parameters of a loss function based on the offset and provide the updated parameters to the neural network to obtain an updated offset.