Adobe Inc.
Methods and systems for automatically correcting image rotation

Abstract:

Systems and methods are disclosed for automatically correcting a rotation of an image. One method comprises receiving an input image to be processed for rotation correction, and providing the input image to a machine learning model configured to predict a rotation angle and an orientation of an image. The machine learning model may comprise a convolutional neural network utilizing one or more rectangle-shaped filters to extract features from the input image. A rotation angle may be determined using a first branch of fully connected layer configured for a regression task of the machine learning model, and an orientation of the image may be determined using a second branch of fully connected layer configured for a classification task of the machine learning model. The input image may then be rotated based on the determined rotation angle and orientation.