/PA_IAA

Primary LanguagePython

Pytorch implementation of "Personality-assisted Multi-task Learning for Generic and Personalized Image Aesthetics Assessment"

  • 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.