search

Wenchao-Du / LIR-for-Unsupervised-IR

This is an implementation for the CVPR2020 paper "Learning Invariant Representation for Unsupervised Image Restoration"
https://arxiv.org/pdf/2003.12769.pdf
100 stars 21 forks source link

Only one encoder in trainer.py #1

Closed petruskkke closed 4 years ago

petruskkke commented 4 years ago

Hi. The paper says there are two encoders (generators) for the clean image and noisy image splitly. However, I can only find one generator 'gen_a' in 'train.py'. I think that it is the generator for noisy image. Where is the defination of the other generator (for clean image)? Thansk in advance.

Wenchao-Du commented 4 years ago

Hi. The paper says there are two encoders (generators) for the clean image and noisy image splitly. However, I can only find one generator 'gen_a' in 'train.py'. I think that it is the generator for noisy image. Where is the defination of the other generator (for clean image)? Thansk in advance.

Yes, the code has been updated for training. Original code has removed the clean encoder branch for testing, you can update the code for customer training.