Skip to content

PyTorch Implementation of TCSVT 2020 "Blind Omnidirectional Image Quality Assessment with Viewport Oriented Graph Convolutional Networks"

License

Notifications You must be signed in to change notification settings

weizhou-geek/VGCN-PyTorch

Folders and files

NameName
Last commit message
Last commit date

Latest commit

Mar 19, 2021
34f4aff · Mar 19, 2021

History

22 Commits
Aug 30, 2020
Mar 19, 2021
Aug 30, 2020
Aug 30, 2020
Aug 26, 2020
Mar 19, 2021
Aug 30, 2020

Repository files navigation

VGCN-PyTorch

Thanks for your attention. In this repo, we provide the codes for the paper [Blind Omnidirectional Image Quality Assessment with Viewport Oriented Graph Convolutional Networks].

Prerequisites

  • scipy==1.2.1
  • opencv_python==4.1.0.25
  • numpy==1.16.4
  • torchvision==0.3.0
  • torch==1.1.0
  • Pillow==6.2.0

Install

To install all the dependencies in prerequisites

Prepare Data

FoV Selection

matlab fov_selection/demo.m

Training

python main.py --root1 cviqd_local_epoch.pth --root2 cviqd_global_epoch.pth --save test

Testing

python main.py --resume cviqd_model.pth --skip_training

Citation

You may cite it in your paper. Thanks a lot.

@article{xu2020blind,
  title={Blind omnidirectional image quality assessment with viewport oriented graph convolutional networks},
  author={Xu, Jiahua and Zhou, Wei and Chen, Zhibo},
  journal={IEEE Transactions on Circuits and Systems for Video Technology},
  year={2020},
  publisher={IEEE}
}

About

PyTorch Implementation of TCSVT 2020 "Blind Omnidirectional Image Quality Assessment with Viewport Oriented Graph Convolutional Networks"

Resources

License

Stars

Watchers

Forks

Releases

No releases published

Packages

No packages published