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

HiddenMarkovModel implements hidden Markov models with Gaussian mixtures as distributions on top of TensorFlow

Get a Grip! - A robotic system for remote clinical environments.

BabelCalib: A Universal Approach to Calibrating Central Cameras. In ICCV (2021)

Official implementation of NPMs: Neural Parametric Models for 3D Deformable Shapes - ICCV 2021

Deep Learning Pipelines for Apache Spark

A project to make Amazon Echo respond to sign language using your webcam

Dynamic Token Normalization Improves Vision Transformers

[CVPR21] LightTrack: Finding Lightweight Neural Network for Object Tracking via One-Shot Architecture Search

Example of semantic segmentation in Keras

The official code of "SCROLLS: Standardized CompaRison Over Long Language Sequences".

Using knowledge-informed machine learning on the PRONOSTIA (FEMTO) and IMS bearing data sets. Predict remaining-useful-life (RUL).

Sinkformers: Transformers with Doubly Stochastic Attention

A PyTorch re-implementation of the paper 'Exploring Simple Siamese Representation Learning'. Reproduced the 67.8% Top1 Acc on ImageNet.

An implementation of the 1. Parallel, 2. Streaming, 3. Randomized SVD using MPI4Py

A PyTorch Implementation of PGL-SUM from "Combining Global and Local Attention with Positional Encoding for Video Summarization", Proc. IEEE ISM 2021

Semi-supervised Implicit Scene Completion from Sparse LiDAR

Digital Twin Mobility Profiling: A Spatio-Temporal Graph Learning Approach

Generic Event Boundary Detection: A Benchmark for Event Segmentation

Inferring Lexicographically-Ordered Rewards from Preferences

Intent parsing and slot filling in PyTorch with seq2seq + attention