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

IDM: An Intermediate Domain Module for Domain Adaptive Person Re-ID,

A Pytorch implementation of "Splitter: Learning Node Representations that Capture Multiple Social Contexts" (WWW 2019).

Predictive Maintenance LSTM

😇A pyTorch implementation of the DeepMoji model: state-of-the-art deep learning model for analyzing sentiment, emotion, sarcasm etc

Our solution for SSN Invente 2021's Hackathon

CDTrans: Cross-domain Transformer for Unsupervised Domain Adaptation

💊 A 3D Generative Model for Structure-Based Drug Design (NeurIPS 2021)

Spherical Confidence Learning for Face Recognition, accepted to CVPR2021.

We will release the code of "ConTNet: Why not use convolution and transformer at the same time?" in this repo

Sound Source Localization for AI Grand Challenge 2021

Collective Multi-type Entity Alignment Between Knowledge Graphs (WWW'20)

Domain Generalization for Mammography Detection via Multi-style and Multi-view Contrastive Learning

Simple reference implementation of GraphSAGE.

PyTorch implementation of popular datasets and models in remote sensing

Official code for "End-to-End Optimization of Scene Layout" -- including VAE, Diff Render, SPADE for colorization (CVPR 2020 Oral)

This repository contains the official MATLAB implementation of the TDA method for reverse image filtering

An implementation on "Curved-Voxel Clustering for Accurate Segmentation of 3D LiDAR Point Clouds with Real-Time Performance"

Implementations of CNNs, RNNs, GANs, etc

PyTorch implementation of ''Background Activation Suppression for Weakly Supervised Object Localization''.

Predicting the duration of arrival delays for commercial flights.