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

 

History

22 Commits
 
 
 
 
 
 
 
 
 
 
 
 
 
 

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