DAN: Unfolding the Alternating Optimization for Blind Super Resolution

Last update: Dec 13, 2022

Related tags

Deep Learning DAN-Basd-on-Openmmlab

Overview

DAN-Basd-on-Openmmlab

DAN: Unfolding the Alternating Optimization for Blind Super Resolution

We reproduce DAN via mmediting based on open-sourced code.

Requirements

PyTorch >= 1.3
mmediting >= 0.9

DataSets

We use DIV2K and Flickr2K as our training datasets. For evaluation of Setting 2, we use DIV2KRK datasets,

Usages

How to run this repo: copy the file to the mmediting workspace and run the program directly based on the commands in mmediting

Copy files to MMEditing workspace.

cd DAN-Basd-on-Openmmlab/
mv ./mmedit/models/restorers/dan.py ${mmediting_workspace}/mmedit/models/restorers/
mv ./mmedit/models/backbones/sr_backbones/dan_net.py ${mmediting_workspace}/mmedit/models/backbones/sr_backbones/
mv ./mmedit/models/common/DANpreprocess.py ${mmediting_workspace}/mmedit/models/common
mv ./configs/restorers/dan ${mmediting_workspace}/configs/restorers/
mv ./tools/data/super-resolution/dan_datasets ${mmediting_workspace}/tools/data/super-resolution/

Modify the configuration file as follows:

pca_matrix_path='${mmediting_workspace}/tools/data/super-resolution/div2k/pca_matrix/pca_aniso_matrix_x4.pth' # your pca_matrix path
# Training
gt_folder='${dataset_workspace}/dataset/DF2K_train_HR_sub' # your train data path
# Testing
lq_folder='${dataset_workspace}/dataset/DIV2KRK/lr_x4' # your test data LR path
gt_folder='${dataset_workspace}/dataset/DIV2KRK/gt' # your test data HR path

Add script to init file, as follows:

modify the mmedit/models/backbones/sr_backbones/__init__.py:

from .dan_net import DAN
# add DAN into __all__ list.

modify the mmedit/models/commons/__init__.py:

from .dan_preprocess import SRMDPreprocessing
# add SRMDreprocessing into __all__ list.

modify the mmedit/models/restorers/__init__.py:

from .dan import DAN
# add DAN into __all__ list.

Training/Test

Before using it, please download and process the dataset and set the path in the configuration file.

Train

# Single GPU
python tools/train.py configs/restorers/dan/dan_setting2.py --work_dir ${YOUR_WORK_DIR}

# Multiple GPUs
./tools/dist_train.sh configs/restorers/dan/dan_setting2.py ${GPU_NUM} --work_dir ${YOUR_WORK_DIR}

Test

# Single GPU
python tools/test.py configs/restorers/dan/dan_setting2.py ${CHECKPOINT_FILE} [--metrics ${METRICS}] [--out ${RESULT_FILE}]

# Multiple GPUs
./tools/dist_test.sh configs/restorers/dan/dan_setting2.py ${CHECKPOINT_FILE} ${GPU_NUM} [--metrics ${METRICS}] [--out ${RESULT_FILE}]

Result

DIV2KRK

The passwds of download links are all 'ta2o'

Method	scale	Datasets	PSNR	Download
DAN	x4	DIV2KRK	27.41	model / test_pkl

DAN: Unfolding the Alternating Optimization for Blind Super Resolution

Related tags

Overview

DAN-Basd-on-Openmmlab

Requirements

DataSets

Usages

Result

DIV2KRK

Owner

AlexZou

All the code and files related to the MI-Lab of UE19CS305 course in sem 5

[ICCV 2021] Deep Hough Voting for Robust Global Registration

Evaluating Cross-lingual Sentence Representations

This project aims at providing a concise, easy-to-use, modifiable reference implementation for semantic segmentation models using PyTorch.

A Pytree Module system for Deep Learning in JAX

Pytorch Implementations of large number classical backbone CNNs, data enhancement, torch loss, attention, visualization and some common algorithms.

The implementation of "Bootstrapping Semantic Segmentation with Regional Contrast".

Official implement of Paper：A deeply supervised image fusion network for change detection in high resolution bi-temporal remote sening images

PyTorch Implementation of SSTNs for hyperspectral image classifications from the IEEE T-GRS paper "Spectral-Spatial Transformer Network for Hyperspectral Image Classification: A FAS Framework."

PyTorch implementation of UPFlow (unsupervised optical flow learning)

Code for the ICCV2021 paper "Personalized Image Semantic Segmentation"

[ACM MM 2021] Multiview Detection with Shadow Transformer (and View-Coherent Data Augmentation)

InsightFace: 2D and 3D Face Analysis Project on MXNet and PyTorch

Final term project for Bayesian Machine Learning Lecture (XAI-623)

Learning to Predict Gradients for Semi-Supervised Continual Learning

Official implementation of the paper Label-Efficient Semantic Segmentation with Diffusion Models

Vanilla and Prototypical Networks with Random Weights for image classification on Omniglot and mini-ImageNet. Made with Python3.

Pytorch implementation for A-NeRF: Articulated Neural Radiance Fields for Learning Human Shape, Appearance, and Pose

Unofficial pytorch implementation of the paper "Dynamic High-Pass Filtering and Multi-Spectral Attention for Image Super-Resolution"

HomoInterpGAN - Homomorphic Latent Space Interpolation for Unpaired Image-to-image Translation