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

A curated list of awesome Model-Based RL resources

Pytorch implementation for the EMNLP 2020 (Findings) paper: Connecting the Dots: A Knowledgeable Path Generator for Commonsense Question Answering

Automated detection of anomalous exoplanet transits in light curve data.

Keras implementation of PersonLab for Multi-Person Pose Estimation and Instance Segmentation.

Official Code for "Constrained Mean Shift Using Distant Yet Related Neighbors for Representation Learning"

Jupyter notebooks for using & learning Keras

Turn based roguelike in python

《Fst Lerning of Temporl Action Proposl vi Dense Boundry Genertor》(AAAI 2020)

Multi-modal co-attention for drug-target interaction annotation and Its Application to SARS-CoV-2

QuakeLabeler is a Python package to create and manage your seismic training data, processes, and visualization in a single place — so you can focus on building the next big thing.

Auxiliary data to the CHIIR paper Searching to Learn with Instructional Scaffolding

LegoDNN: a block-grained scaling tool for mobile vision systems

Investigating automatic navigation towards standard US views integrating MARL with the virtual US environment developed in CT2US simulation

ML-PersonalWork - Big assignment PersonalWork in Machine Learning, 2021 autumn BUAA.

Code for A Volumetric Transformer for Accurate 3D Tumor Segmentation

(CVPR 2022 - oral) Multi-View Depth Estimation by Fusing Single-View Depth Probability with Multi-View Geometry

NLP made easy

Deep motion generator collections

Harmonious Textual Layout Generation over Natural Images via Deep Aesthetics Learning

Code for ICLR 2020 paper "VL-BERT: Pre-training of Generic Visual-Linguistic Representations".