If you find our work is useful, pleaes cite our paper:
@ARTICLE{Li2020Personality,
author={L. {Li} and H. {Zhu} and S. {Zhao} and G. {Ding} and W. {Lin}},
journal={IEEE Trans. Image Process.},
title={Personality-assisted Multi-task Learning for Generic and Personalized Image Aesthetics Assessment},
year={2020},
volume={29},
number={},
pages={3898-3910},
}

Model_test.py for general aesthetic model testing.

Model_test_AVA.py for general aesthetic model testing on AVA database.

Model_train.py for general aesthetic model training.

Model_fineTune.py for personalized aesthetic model fine-tuning. Pre-trained model can be downloaded here

If you want to train our general model, you need download PsychoFlickr, AVA and AADB databases.