Home

Awesome

DEIQT

Checkpoints, logs and source code for AAAI-23 paper 'Data-Efficient Image Quality Assessment with Attention-Panel Decoder'

Updates

To-Dos

Dependencies

Usage

Pre-requisition

Weights

Download the Pre-trained ViT-S[224] weights from DeiT III

WandB

This project use WandB to log information and report. Remember to adjust the code in main.py to suit your research.

Training

CUDA_VISIBLE_DEVICES=0,1,2,3 OMP_NUM_THREADS=1 torchrun --nnodes 1 --nproc_per_node 4 --master_port 49935  main.py \
--cfg [CONFIG_PATH] \
--data-path [YOUR_DATA_PATH] \
--output [LOG_PATH] \
--tag [REMARK_TAG] \
--repeat \
--rnum [TARGET_REPEAT_NUM]

Citing DEIQT

If you find this project helpful in your research, please consider citing our papers:

@inproceedings{qin2023deiqt,
  title={Data-Efficient Image Quality Assessment with Attention-Panel Decoder},
  author={Guanyi Qin and Runze Hu and Yutao Liu and Xiawu Zheng and Haotian Liu and Xiu Li and Yan Zhang},
  booktitle={Proceedings of the AAAI Conference on Artificial Intelligence (AAAI)},
  year={2023}
}

Acknowledgement

We borrowed some parts from the following open-source projects:

Many thanks to them.