mc360iqa's Introduction

MC360IQA

MC360IQA: A Multi-channel CNN for Blind 360-degree Image Quality Assessment

Usage

If you want to train the code on your database (e.g. CVIQ database ):

First, prepare the database

cd equi2cubic
ConvertCVIQtoCubic.m

Then

CUDA_VISIBLE_DEVICES=0 python train.py \
--num_epochs 10 \
--batch_size 40 \
--database CVIQ \
--data_dir /DATA/CVIQcubic \
--filename_train CVIQ/CVIQ_train.csv \
--filename_test CVIQ/CVIQ_test.csv \
--snapshot  /DATA/ModelFolder/VRIQA \
--cross_validation_index 1

If you want to test the trained model on the test set:

CUDA_VISIBLE_DEVICES=1 python test.py \
--database CVIQ \
--data_dir /DATA/CVIQcubic \
--filename_test CVIQ/CVIQ_test.csv \
--snapshot  /DATA/ModelFolder/VRIQA/CVIQ/1/CVIQ.pkl

If you just want to evaluate the quality of an equirectangular image:

CUDA_VISIBLE_DEVICES=0 python test_on_equirectangular.py \
--filename images/1.png \
--snapshot  /DATA/ModelFolder/VRIQA/CVIQ/1/CVIQ.pkl

Model

You can download the trained model via:

CVIQ: google drive baidu yun 提取码：5muh

OIQA: google drive baidu yun 提取码：39we

We recommend you to use the model trained on the OIQA database since it is more robust.

Citation

If you find this code is useful for your research, please cite:

@article{sun2019mc360iqa,
  title={MC360IQA: A Multi-channel CNN for Blind 360-degree Image Quality Assessment},
  author={Sun, Wei and Min, Xiongkuo and Zhai, Guangtao and Gu, Ke and Duan, Huiyu and Ma, Siwei},
  journal={IEEE Journal of Selected Topics in Signal Processing},
  volume={14},
  number={1},
  pages={64-77},
  year={2020},
  publisher={IEEE}
}