Automatic orientation correction for captured images

A method of automatic orientation correction for captured images, comprising:

• receiving, by a device, an image of a document, the image depicting a reference feature associated with the document, the reference feature including at least one of:
• a face of a person,
• a machine-readable code, or
• a text field;
• selecting, by the device, a machine learning model, from a plurality of machine learning models, based at least on one of the reference feature or a document type associated with the document;
• identifying, by the device and using the selected machine learning model, a rotational angle of the reference feature as depicted in the image based on comparing the reference feature as depicted in the image to one or more orientation parameters of the reference feature associated with a display orientation associated with the document;
• rotating, by the device, the image of the document by an angle to obtain an orientated image of the document, the angle being based on the rotational angle of the reference feature as depicted in the image;
• providing, by the device, the orientated image of the document for display;
• receiving, by the device, feedback information indicating whether the oriented image has been accepted;
• transmitting, by the device and based on the feedback information, the oriented image or the image of the document to a server device as part of an authentication procedure;
• and
• re-training, by the device and based on the feedback information, the selected machine learning model.

A non-transitory computer-readable medium storing a set of instructions, the set of instructions comprising:

• one or more instructions that, when executed by one or more processors of a device, cause the device to:
• obtain, via a user device, an image of an identification document, the identification document including a facial image depicting a face of a user associated with the identification document;
• select a machine learning model, from a plurality of machine learning models, based on at least one of one or more reference features associated with the identification document or a document type associated with the identification document,
• wherein the machine learning model is a facial recognition model that is trained to identify rotational angles of faces, and
• wherein one of the one or more reference features is associated with a facial image;
• input, to the selected machine learning model, the image of the identification document;
• obtain, from the selected machine learning model, an indication of a rotational angle of the face depicted in the facial image based on inputting, to the selected machine learning model, the image of the identification document;
• rotate the image of the identification document by an angle to obtain an orientated image of the identification document, the angle being based on the rotational angle of the face and a reference angle associated with a display orientation of the identification document;
• provide the orientated image of the identification document for display;
• receive feedback information indicating whether the oriented image has been accepted;
• transmit, based on the feedback information, the oriented image or the image of the identification document to a server device as part of an authentication procedure related to the user device;
• and
• re-train, based on the feedback information, the selected machine learning model.