Skip to content

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

Notifications You must be signed in to change notification settings

narthchin/DEIQT

Folders and files

NameName
Last commit message
Last commit date

Latest commit

 

History

4 Commits
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 

Repository files navigation

DEIQT

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

Updates

  • [06/2023] We released the source code of 'DEIQT', check the code on GitHub
  • [04/2023] We released our work 'DEIQT', the paper is now on Arxiv.

To-Dos

  • Checkpoints & Logs
  • Initialization

Dependencies

  • Python
  • PyTorch

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.

About

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

Resources

Stars

Watchers

Forks

Releases

No releases published

Packages

No packages published

Languages