Pytorch implementation of "ARM: Any-Time Super-Resolution Method"

Last update: Nov 24, 2022

Related tags

Overview

ARM-Net

Dependencies

Python 3.6
Pytorch 1.7

Results

Train

Data preprocessing

cd data_scripts
python extract_subimages_test.py
python data_augmentation.py
python generate_mod_LR_bic.py
python extract_subimages_train.py

Modify the configuration file (options/*.yml)

# train_us_fsrcnn.yml
is_train: True

# train_us_carn.yml
is_train: True

# train_us_srresnet.yml
is_train: True

Run training scripts

python main.py -opt options/train_us_fsrcnn.yml
python main.py -opt options/train_us_carn.yml
python main.py -opt options/train_us_srresnet.yml

Test

Modify the configuration file (options/*.yml)

# train_us_fsrcnn.yml
path:
  pretrain_model_G: ckpt/arm-fsrcnn.pth
  resume_state: ckpt/arm-fsrcnn.state
is_train: False
is_test: True

# train_us_carn.yml
path:
  pretrain_model_G: ckpt/arm-carn.pth
  resume_state: ckpt/arm-carn.state
is_train: False
is_test: True

# train_us_srresnet.yml
path:
  pretrain_model_G: ckpt/arm-srresnet.pth
  resume_state: ckpt/arm-srresnet.state
is_train: False
is_test: True

Run test scripts

python main.py -opt options/train_us_fsrcnn.yml
python main.py -opt options/train_us_carn.yml
python main.py -opt options/train_us_srresnet.yml

A PyTorch Extension: Tools for easy mixed precision and distributed training in Pytorch

This repository holds NVIDIA-maintained utilities to streamline mixed precision and distributed training in Pytorch. Some of the code here will be included in upstream Pytorch eventually. The intention of Apex is to make up-to-date utilities available to users as quickly as possible.

6.9k Jan 3, 2023

Objective of the repository is to learn and build machine learning models using Pytorch. 30DaysofML Using Pytorch

30 Days Of Machine Learning Using Pytorch Objective of the repository is to learn and build machine learning models using Pytorch. List of Algorithms

119 Nov 24, 2022

Pretrained SOTA Deep Learning models, callbacks and more for research and production with PyTorch Lightning and PyTorch

1.4k Jan 1, 2023

Amazon Forest Computer Vision: Satellite Image tagging code using PyTorch / Keras with lots of PyTorch tricks

Amazon Forest Computer Vision Satellite Image tagging code using PyTorch / Keras Here is a sample of images we had to work with Source: https://www.ka

360 Dec 10, 2022

The Incredible PyTorch: a curated list of tutorials, papers, projects, communities and more relating to PyTorch.

This is a curated list of tutorials, projects, libraries, videos, papers, books and anything related to the incredible PyTorch. Feel free to make a pu

9.2k Jan 2, 2023

Amazon Forest Computer Vision: Satellite Image tagging code using PyTorch / Keras with lots of PyTorch tricks

Amazon Forest Computer Vision Satellite Image tagging code using PyTorch / Keras Here is a sample of images we had to work with Source: https://www.ka

359 Jan 5, 2023

A bunch of random PyTorch models using PyTorch's C++ frontend

PyTorch Deep Learning Models using the C++ frontend Gettting started Clone the repo 1. https://github.com/mrdvince/pytorchcpp 2. cd fashionmnist or

0 Jul 13, 2021

PyTorch Autoencoders - Implementing a Variational Autoencoder (VAE) Series in Pytorch.

PyTorch Autoencoders Implementing a Variational Autoencoder (VAE) Series in Pytorch. Inspired by this repository Model List check model paper conferen

8 Nov 21, 2022

PyTorch-LIT is the Lite Inference Toolkit (LIT) for PyTorch which focuses on easy and fast inference of large models on end-devices.

PyTorch-LIT PyTorch-LIT is the Lite Inference Toolkit (LIT) for PyTorch which focuses on easy and fast inference of large models on end-devices. With

157 Dec 11, 2022

Comments

What does ARM-L/ARM-M/ARM-S mean?
It is a interesting work. I have some questions about your paper.

In your work. When inference, my understanding is to build edge table at first, then use this table when inference. But How to decide which width network a patch choose? The others methods in Figure 7 is clear. But I can't know how ARM-FSRCNN choose width. It is confuse me (

In Table 1&2, ARM-L/ARM-M/ARM-S means: fixed width no matter what the input patch is, right?

About ARM-L/ARM-M/ARM-S, If just fix width. May be you should compare with ClassSR use your width choose policy insted of fixing width?

By the way. In table 1 (row: Module FSRCNN, column: FLOPS) is 0%, may be error. :)

Many questions from me. Thanks!
opened by wangqiim 2
Where is the defination of "ProgressBar()" in "utils.util" which is imported in extract_subimages_train.py ?

Where is the defination of "ProgressBar()" in "utils.util" ? The "ProgressBar" is imported in extract_subimages_train.py.

opened by huang-hz 1
New Super-Resolution Benchmarks
Hello,

MSU Graphics & Media Lab Video Group has recently launched two new Super-Resolution Benchmarks.

Video Upscalers Benchmark: Quality Enhancement determines the best upscaling methods for increasing video resolution and improving visual quality.

Super-Resolution for Video Compression benchmark aims to test Super-Resolution methods on compressed videos and select the best model for each video codec standard.

If you are interested in participating, you can add your algorithm following the submission steps:

Submit for Video Upscalers Benchmark: Quality Enhancement

Submit for Super-Resolution for Video Compression benchmark

We would be grateful for your feedback on our work!
opened by EvgeneyBogatyrev 0
How to get Fig.4. (a) 0.85 ?

In Fig.4.(a), I find different models will cause different PSNR lookup results. How to get Spearman correlation (0.85) coefficient picture (Fig.4.(a)). What model do you use to generate psnr values. (complex model will cause more pricise PSNR value)

opened by wangqiim 0

Pytorch implementation of "ARM: Any-Time Super-Resolution Method"

Related tags

Overview

ARM-Net

Dependencies

Results

Train

Data preprocessing

Modify the configuration file (options/*.yml)

Run training scripts

Test

Modify the configuration file (options/*.yml)

Run test scripts

You might also like...

A PyTorch Extension: Tools for easy mixed precision and distributed training in Pytorch

Objective of the repository is to learn and build machine learning models using Pytorch. 30DaysofML Using Pytorch

Pretrained SOTA Deep Learning models, callbacks and more for research and production with PyTorch Lightning and PyTorch

Amazon Forest Computer Vision: Satellite Image tagging code using PyTorch / Keras with lots of PyTorch tricks

The Incredible PyTorch: a curated list of tutorials, papers, projects, communities and more relating to PyTorch.

Amazon Forest Computer Vision: Satellite Image tagging code using PyTorch / Keras with lots of PyTorch tricks

A bunch of random PyTorch models using PyTorch's C++ frontend

PyTorch Autoencoders - Implementing a Variational Autoencoder (VAE) Series in Pytorch.

PyTorch-LIT is the Lite Inference Toolkit (LIT) for PyTorch which focuses on easy and fast inference of large models on end-devices.

Comments

What does ARM-L/ARM-M/ARM-S mean?

Where is the defination of "ProgressBar()" in "utils.util" which is imported in extract_subimages_train.py ?

New Super-Resolution Benchmarks

How to get Fig.4. (a) 0.85 ?

Releases(v1.0.0)

v1.0.0(Sep 28, 2022)

Owner

Bohong Chen

Everything you need to know about NumPy( Creating Arrays, Indexing, Math,Statistics,Reshaping).

CountDown to New Year and shoot fireworks

The Implicit Bias of Gradient Descent on Generalized Gated Linear Networks

Medical Insurance Cost Prediction using Machine earning

Implementation of our paper "Video Playback Rate Perception for Self-supervised Spatio-Temporal Representation Learning".

🏖 Keras Implementation of Painting outside the box

Code for "Neural Body: Implicit Neural Representations with Structured Latent Codes for Novel View Synthesis of Dynamic Humans" CVPR 2021 best paper candidate

GAN Image Generator and Characterwise Image Recognizer with python

Extracting and filtering paraphrases by bridging natural language inference and paraphrasing

PyTorch Implementation of Small Lesion Segmentation in Brain MRIs with Subpixel Embedding (ORAL, MICCAIW 2021)

Code for ACL2021 paper Consistency Regularization for Cross-Lingual Fine-Tuning.

Implementation of " SESS: Self-Ensembling Semi-Supervised 3D Object Detection" (CVPR2020 Oral)

Code examples and benchmarks from the paper "Understanding Entropy Coding With Asymmetric Numeral Systems (ANS): a Statistician's Perspective"

An implementation of IMLE-Net: An Interpretable Multi-level Multi-channel Model for ECG Classification

Light-weight network, depth estimation, knowledge distillation, real-time depth estimation, auxiliary data.

Large-Scale Unsupervised Object Discovery

Multiband spectro-radiometric satellite image analysis with K-means cluster algorithm

Learning Confidence for Out-of-Distribution Detection in Neural Networks

Implementation of NÜWA, state of the art attention network for text to video synthesis, in Pytorch

This code is the implementation of the paper "Coherence-Based Distributed Document Representation Learning for Scientific Documents".