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.