Machine learning assisted image prediction

A device, comprising:

• an infrared image sensor;
• at least one processor;
• and
• a memory including instructions that, when executed by the at least one processor, cause the at least one processor to:
• capture, by the infrared image sensor, an infrared image of a subject;
• obtain a reference RGB image of the subject;
• provide the infrared image and the reference RGB image to a machine learning model, the machine learning model having been trained to output predicted RGB images based on infrared images and corresponding RGB images of subjects;
• and
• provide a predicted RGB image of the subject based on an output of the machine learning model.

A non-transitory computer-readable medium comprising instruction which, when executed by one or more processors, cause the one or more processors to:

• obtain an infrared image of a subject;
• obtain a reference color image of the subject;
• select a machine learning model from plural machine learning models based on an ambient lux level;
• provide the infrared image and the reference color image to the selected machine learning model, the selected machine learning model having been trained to output predicted color images based on infrared images and reference color images of subjects;
• and
• provide a predicted color image of the subject based on an output of the selected machine learning model.