/BAID

Primary LanguagePython

Official repository for CVPR2023 paper: "Towards Artistic Image Aesthetics Assessment: a Large-scale Dataset and a New Method"

Dataset

  • Clone the repository
git clone https://github.com/Dreemurr-T/BAID.git
cd BAID/
  • Install the necessary dependencies using:
pip install pandas
pip install tqdm
  • Download the dataset using:
python downloading_script/download.py

The images will be saved to images/ folder.

Since it might be slow when downloading the images, we provide alternatives to obtain the dataset:

  • Baidu Netdisk: Link, Code: 9y91
  • Google Drive: Coming soon

Ground-truth labels of the dataset can be found in the dataset folder.

Code

Requirements

  • Python >= 3.8
  • Pytorch >= 1.12.0
  • Torchvision >= 0.13.0

Other dependencies can be installed with:

pip install -r requirements.txt

Pretraining

  • Download the BAID dataset and place the images in the images/ folder
  • Preprocess the data using:
python pretraining_utils/pretrain_mani.py
  • Pretrain the ResNet50 backbone using:
python pretraining.py

The whole pretraining process takes about 2 days on a single RTX3090. We provide our pretrained weights at Drive.

Training

For training on BAID, use:

python train.py

Checkpoints will be save to checkpoint/SAAN folder.

Testing

For testing on BAID, download the pretrained weights from Drive, place the checkpoint in checkpoint/BAID

Then use:

python test.py

License

The dataset is licensed under CC BY-NC-ND 4.0

Acknowledgement

The code borrowed from pytorch-AdaIN and Non-local_pytorch.

Citation

If you find our work useful, please cite our work as:

@InProceedings{Yi_2023_CVPR,
    author    = {Yi, Ran and Tian, Haoyuan and Gu, Zhihao and Lai, Yu-Kun and Rosin, Paul L.},
    title     = {Towards Artistic Image Aesthetics Assessment: A Large-Scale Dataset and a New Method},
    booktitle = {Proceedings of the IEEE/CVF Conference on Computer Vision and Pattern Recognition (CVPR)},
    month     = {June},
    year      = {2023},
    pages     = {22388-22397}
}