Code for the paper "Location-aware Single Image Reflection Removal"

Last update: Dec 8, 2022

Overview

Location-aware Single Image Reflection Removal

The shown images are provided by the datasets from IBCLN, ERRNet, SIR² and the Internet images.

The code and pretrained model for our paper: Location-aware Single Image Reflection Removal [Arxiv Preprint]

Prerequisites

Our code has been tested under the following platform and environment:

Ubuntu. CPU or NVIDIA GPU + CUDA, CuDNN
Python 3.7.3, Pytorch 1.2.0
Requirements: numpy, tqdm, Pillow, dominate, scikit-image

Setup

Clone or Download this repo
$ cd Location-aware-SIRR
$ mkdir model
Download the pretrained model here
Move the downloaded model(model.pth) to ./model folder

Usage

The example test images are provided in ./test_images/blend folder
If you have ground truth blackground images, put them into ./test_images/transmission folder ( Note that the same pair of images need to be named the same ).
Run python3 inference.py
The inference results are in the ./results folder

Citation

If you find our work helpful to your research, please cite our paper.

@article{dong2020location,
  author = {Zheng Dong and Ke Xu and Yin Yang and Hujun Bao and Weiwei Xu and Rynson W.H. Lau},
  title = {Location-aware Single Image Reflection Removal},
  journal={ArXiv},
  volume={abs/2012.07131},
  year = {2020}
}

You might also like...

PyTorch code for the paper: FeatMatch: Feature-Based Augmentation for Semi-Supervised Learning

FeatMatch: Feature-Based Augmentation for Semi-Supervised Learning This is the PyTorch implementation of our paper: FeatMatch: Feature-Based Augmentat

43 Nov 19, 2022

Code for the paper A Theoretical Analysis of the Repetition Problem in Text Generation

A Theoretical Analysis of the Repetition Problem in Text Generation This repository share the code for the paper "A Theoretical Analysis of the Repeti

37 Nov 21, 2022

Code for our ICASSP 2021 paper: SA-Net: Shuffle Attention for Deep Convolutional Neural Networks

SA-Net: Shuffle Attention for Deep Convolutional Neural Networks (paper) By Qing-Long Zhang and Yu-Bin Yang [State Key Laboratory for Novel Software T

199 Jan 8, 2023

Open source repository for the code accompanying the paper 'Non-Rigid Neural Radiance Fields Reconstruction and Novel View Synthesis of a Deforming Scene from Monocular Video'.

Non-Rigid Neural Radiance Fields This is the official repository for the project "Non-Rigid Neural Radiance Fields: Reconstruction and Novel View Synt

296 Dec 29, 2022

Comments

Training source code

Hello, teacher, The results of your experiment are very impressive on real world dataset. I would like to reproduce the training process to have further research.But there is only evaluation code in github, could you provide the training source code? Thank you so much!

opened by GYC124 0
Training source code

Hello, teacher, The results of your experiment are very impressive on real world dataset. I would like to reproduce the training process to have further research.But there is only evaluation code in github, could you provide the training source code? Thank you so much!

opened by ray221183 0

Run data loader in main thread

This fixes the runtime error below, when running inference.py

RuntimeError: 
        An attempt has been made to start a new process before the
        current process has finished its bootstrapping phase.

        This probably means that you are not using fork to start your
        child processes and you have forgotten to use the proper idiom
        in the main module:

            if __name__ == '__main__':
                freeze_support()
                ...

        The "freeze_support()" line can be omitted if the program
        is not going to be frozen to produce an executable.

opened by andreas128 0

Request issues

Hello, teacher. I downloaded the code and model you provided on the Internet pth。 I want to reproduce the teacher's experiment first, but I'm running information There are the following errors in py (with photos). Do you know the reason, teacher? Thank the teacher for taking time out of his busy schedule to answer the students. Thank you very much!!

opened by 8267134 2

Code for the paper "Location-aware Single Image Reflection Removal"

Related tags

Overview

Location-aware Single Image Reflection Removal

Prerequisites

Setup

Usage

Citation

You might also like...

PyTorch code for the paper: FeatMatch: Feature-Based Augmentation for Semi-Supervised Learning

Code for the paper A Theoretical Analysis of the Repetition Problem in Text Generation

Code for our ICASSP 2021 paper: SA-Net: Shuffle Attention for Deep Convolutional Neural Networks

Open source repository for the code accompanying the paper 'Non-Rigid Neural Radiance Fields Reconstruction and Novel View Synthesis of a Deforming Scene from Monocular Video'.

Code for the Shortformer model, from the paper by Ofir Press, Noah A. Smith and Mike Lewis.

PyTorch code for ICLR 2021 paper Unbiased Teacher for Semi-Supervised Object Detection

Official code for paper "Optimization for Oriented Object Detection via Representation Invariance Loss".

Code for our CVPR 2021 paper "MetaCam+DSCE"

Code for our CVPR2021 paper coordinate attention

Comments

Training source code

Training source code

Run data loader in main thread

Request issues

Owner

This is the official source code for SLATE. We provide the code for the model, the training code, and a dataset loader for the 3D Shapes dataset. This code is implemented in Pytorch.

Code for paper ECCV 2020 paper: Who Left the Dogs Out? 3D Animal Reconstruction with Expectation Maximization in the Loop.

TensorFlow code for the neural network presented in the paper: "Structural Language Models of Code" (ICML'2020)

Code for the prototype tool in our paper "CoProtector: Protect Open-Source Code against Unauthorized Training Usage with Data Poisoning".

Code to use Augmented Shapiro Wilks Stopping, as well as code for the paper "Statistically Signifigant Stopping of Neural Network Training"

Code for our method RePRI for Few-Shot Segmentation. Paper at http://arxiv.org/abs/2012.06166

Code for ACM MM 2020 paper "NOH-NMS: Improving Pedestrian Detection by Nearby Objects Hallucination"

Official TensorFlow code for the forthcoming paper

This is the code for the paper "Contrastive Clustering" (AAAI 2021)

Code for the paper Learning the Predictability of the Future