The invention relates to an attention mechanism-based image aesthetics quality evaluation method, and belongs to the technical field of computer vision. The method comprises the steps of firstly processing training data, then designing a network structure model, adopting a lightweight deep network as a backbone network, and integrating the backbone network into an attention mechanism module; designing a loss function for training the network based on a data equalization thought; finally, training a network structure model by using the processed training data to obtain a network model capable of automatically evaluating the aesthetic quality of an image; and performing aesthetic scoring on the input picture based on the model, and applying the model to shooting to assist a user to shoot a more beautiful picture in real time. Compared with the prior art, the network structure model adopted by the invention can more effectively extract the characteristics of images, the adopted loss function greatly enhances the data learning ability of the model. Compared with other methods, the accuracy is improved, and the parameter quantity of the model is reduced.
Attention mechanism-based image aesthetics quality evaluation method