Method and system for detecting and correcting an orientation of an image

A system for correcting an orientation of an input image, the system comprising:

• an image processing device comprising at least one processor and a computer-readable medium storing instructions that, when executed by the at least one processor, cause the at least one processor to perform operations comprising:
• generating a set of images of an environment corresponding to the input image;
• determining a region and a label corresponding to each of one or more objects in each of the set of images using an object detection and labelling model;
• determining a context for each of the one or more objects in each of the set of images with respect to an overall context of the environment, using a context generation model, based on the label for each of the one or more objects;
• estimating an orientation score for each of the set of images based on a correlation between the region and the label of each of the one or more objects in each of the set of images, and the context of each of the one or more objects in each of the set of images;
• and
• correcting the orientation of the input image based on the estimated orientation score for each of the set of images.

A non-transitory computer-readable medium storing computer-executable instructions for:

• generating a set of images of an environment corresponding to an input image;
• determining a region and a label corresponding to each of one or more objects in each of the set of images using an object detection and labelling model;
• determining a context for each of the one or more objects in each of the set of images with respect to an overall context of the environment, using a context generation model, based on the label for each of the one or more objects;
• estimating an orientation score for each of the set of images based on a correlation between the region and the label of each of the one or more objects in each of the set of images, and the context of each of the one or more objects in each of the set of images;
• and
• correcting an orientation of the input image based on the estimated orientation score for each of the set of images.