[CVPRW 2021] Code for Region-Adaptive Deformable Network for Image Quality Assessment

Overview

RADN

[CVPRW 2021] Code for Region-Adaptive Deformable Network for Image Quality Assessment

[Paper on arXiv]

Overview

Update

[2021/5/7] add codes for WResNet (our baseline).

Instruction

  1. run mkdir.sh to create necessary directories.

  2. use sh train.sh or sh test.sh to train or test the model. You can also change the options in the shell files as you like.

The pretrained models can be found at this URL.

Please note that the performance on the challenge leaderboard is obtained by ensembling and the checkpoint above is for the single model.

Performance

Scatter Plots

Attention Maps

Acknowledgment

The codes borrow heavily from WaDIQaM implemented by Dingquan Li and we really appreciate it.

Citation

If you find our work or code helpful for your research, please consider to cite:

@inproceedings{RADN2021ntire, 
title={Region-Adaptive Deformable Network for Image Quality Assessment}, 
author={Shuwei Shi and Qingyan Bai and Mingdeng Cao and Weihao Xia and Jiahao Wang and Yifan Chen and Yujiu Yang}, 
booktitle={IEEE/CVF Conference on Computer Vision and Pattern Recognition Workshops}, 
year={2021} 
}
You might also like...
No-reference Image Quality Assessment(NIQA) Algorithms (BRISQUE, NIQE, PIQE, RankIQA, MetaIQA)
No-reference Image Quality Assessment(NIQA) Algorithms (BRISQUE, NIQE, PIQE, RankIQA, MetaIQA)

No-Reference Image Quality Assessment Algorithms No-reference Image Quality Assessment(NIQA) is a task of evaluating an image without a reference imag

[ICCV 2021] Group-aware Contrastive Regression for Action Quality Assessment
[ICCV 2021] Group-aware Contrastive Regression for Action Quality Assessment

CoRe Created by Xumin Yu*, Yongming Rao*, Wenliang Zhao, Jiwen Lu, Jie Zhou This is the PyTorch implementation for ICCV paper Group-aware Contrastive

Accurate 3D Face Reconstruction with Weakly-Supervised Learning: From Single Image to Image Set (CVPRW 2019). A PyTorch implementation.
Accurate 3D Face Reconstruction with Weakly-Supervised Learning: From Single Image to Image Set (CVPRW 2019). A PyTorch implementation.

Accurate 3D Face Reconstruction with Weakly-Supervised Learning: From Single Image to Image Set —— PyTorch implementation This is an unofficial offici

MagFace: A Universal Representation for Face Recognition and Quality Assessment
MagFace: A Universal Representation for Face Recognition and Quality Assessment

MagFace MagFace: A Universal Representation for Face Recognition and Quality Assessment in IEEE Conference on Computer Vision and Pattern Recognition

Implementation of temporal pooling methods studied in [ICIP'20] A Comparative Evaluation Of Temporal Pooling Methods For Blind Video Quality Assessment

Implementation of temporal pooling methods studied in [ICIP'20] A Comparative Evaluation Of Temporal Pooling Methods For Blind Video Quality Assessment

MRQy is a quality assurance and checking tool for quantitative assessment of magnetic resonance imaging (MRI) data.
MRQy is a quality assurance and checking tool for quantitative assessment of magnetic resonance imaging (MRI) data.

Front-end View Backend View Table of Contents Description Prerequisites Running Basic Information Measurements User Interface Feedback and usage Descr

PyTorch version of the paper 'Enhanced Deep Residual Networks for Single Image Super-Resolution' (CVPRW 2017)
PyTorch version of the paper 'Enhanced Deep Residual Networks for Single Image Super-Resolution' (CVPRW 2017)

About PyTorch 1.2.0 Now the master branch supports PyTorch 1.2.0 by default. Due to the serious version problem (especially torch.utils.data.dataloade

Official PyTorch implementation of
Official PyTorch implementation of "IntegralAction: Pose-driven Feature Integration for Robust Human Action Recognition in Videos", CVPRW 2021

IntegralAction: Pose-driven Feature Integration for Robust Human Action Recognition in Videos Introduction This repo is official PyTorch implementatio

CVPRW 2021: How to calibrate your event camera
CVPRW 2021: How to calibrate your event camera

E2Calib: How to Calibrate Your Event Camera This repository contains code that implements video reconstruction from event data for calibration as desc

Comments
  • Request to answer questions

    Request to answer questions

    In test_val.py, the pre-trained model is "WResNet-lr=0.0001-bs=2-10". However, your released pre-trained in google drive is "WResNet-lr=0.0001-bs=2". What's the difference between these two pre-trained models.

    opened by happycaoyue 3
  • The following bugs appear in Deform model during debugging

    The following bugs appear in Deform model during debugging

    不太理解这部分的调用,错误也发生在这部分 I don't understand this part of the call,error also occur here torch.ops.torchvision.deform_conv2d() 错误信息如下 The error message is as follows

    RuntimeError: torchvision::deform_conv2d() Expected a value of type 'Tensor' for argument 'bias' but instead found type 'int'.
    Position: 4
    Value: 1
    Declaration: torchvision::deform_conv2d(Tensor input, Tensor weight, Tensor offset, Tensor mask, Tensor bias, int stride_h, int stride_w, int pad_h, int pad_w, int dilation_h, int dilation_w, int groups, int offset_groups, bool use_mask) -> (Tensor)
    Cast error details: Unable to cast Python instance to C++ type (compile in debug mode for details)
    

    Looking forward to your reply

    opened by Oraclexu 2
  • 重新训练的时候发现RADN的结果都是nan,请问是什么原因?

    重新训练的时候发现RADN的结果都是nan,请问是什么原因?

    用RADN进行训练测试的结果如下: Validation Results - Epoch: 6 SROCC: nan KROCC: nan PLCC: nan RMSE: 119.7862 MAE: 93.5760 OR: 0.00% 而用WResNet的结果就在0.75左右,是不是RADN代码有问题?

    opened by lllllllllllll-llll 1
  • 关于contrastive pretraining strategy的提问

    关于contrastive pretraining strategy的提问

    作者您好,很感激您开源了你们这份代码,但是在其中有一些没明白的点,希望您能解惑一下: 论文中提到有利用孪生模型利用MOS分之间隐含的rank信息而不是单纯对MOS分做回归,会生成一波preference probabillity与预测分数si,然后与实际的prefenrece probabillity和实际分数算cross entrophy,但是在代码中好像没有看到这个。想知道在这个策略是用在最后来验证WResNet的性能还是用作WResNet的pretraining呢?

    opened by TowardsXY 0
Owner
IIGROUP
The Intelligent Interaction Group at Tsinghua University
IIGROUP
Implementation of Deformable Attention in Pytorch from the paper "Vision Transformer with Deformable Attention"

Deformable Attention Implementation of Deformable Attention from this paper in Pytorch, which appears to be an improvement to what was proposed in DET

Phil Wang 128 Dec 24, 2022
Cross-Image Region Mining with Region Prototypical Network for Weakly Supervised Segmentation

Cross-Image Region Mining with Region Prototypical Network for Weakly Supervised Segmentation The code of: Cross-Image Region Mining with Region Proto

LiuWeide 16 Nov 26, 2022
Source code for paper "Deep Superpixel-based Network for Blind Image Quality Assessment"

DSN-IQA Source code for paper "Deep Superpixel-based Network for Blind Image Quality Assessment" Requirements Python >=3.8.0 Pytorch >=1.7.1 Usage wit

null 7 Oct 13, 2022
Some code of the implements of Geological Modeling Using 3D Pixel-Adaptive and Deformable Convolutional Neural Network

3D-GMPDCNN Geological Modeling Using 3D Pixel-Adaptive and Deformable Convolutional Neural Network PyTorch implementation of "Geological Modeling Usin

null 5 Nov 21, 2022
Code for Dual Contrastive Learning for Unsupervised Image-to-Image Translation, NTIRE, CVPRW 2021.

arXiv Dual Contrastive Learning Adversarial Generative Networks (DCLGAN) We provide our PyTorch implementation of DCLGAN, which is a simple yet powerf

null 119 Dec 4, 2022
Official PyTorch implementation of the paper "Recycling Discriminator: Towards Opinion-Unaware Image Quality Assessment Using Wasserstein GAN", accepted to ACM MM 2021 BNI Track.

RecycleD Official PyTorch implementation of the paper "Recycling Discriminator: Towards Opinion-Unaware Image Quality Assessment Using Wasserstein GAN

Yunan Zhu 23 Nov 5, 2022
[CVPR 2021] Region-aware Adaptive Instance Normalization for Image Harmonization

RainNet — Official Pytorch Implementation Region-aware Adaptive Instance Normalization for Image Harmonization Jun Ling, Han Xue, Li Song*, Rong Xie,

null 130 Dec 11, 2022
Code for the paper SphereRPN: Learning Spheres for High-Quality Region Proposals on 3D Point Clouds Object Detection, ICIP 2021.

SphereRPN Code for the paper SphereRPN: Learning Spheres for High-Quality Region Proposals on 3D Point Clouds Object Detection, ICIP 2021. Authors: Th

Thang Vu 15 Dec 2, 2022
Lightweight Face Image Quality Assessment

LightQNet This is a demo code of training and testing [LightQNet] using Tensorflow. Uncertainty Losses: IDQ loss PCNet loss Uncertainty Networks: Mobi

Kaen 5 Nov 18, 2022