search

aitorzip / PyTorch-SRGAN

A modern PyTorch implementation of SRGAN
https://arxiv.org/abs/1609.04802
GNU General Public License v3.0
359 stars 92 forks source link

How to use SRGAN with black and white images #8

Open lilachperry opened 6 years ago

lilachperry commented 6 years ago

Hi,

Thx for sharing the code. I'm trying to use the SRGAN with black and white images. The first problem I ran into was that VGG expects 3 channels. The easiest solution is to repeat my black and white data x3 along the channel axis. Do you have any experience with b/w data? maybe I need to do some pre-processing ?

Thanks, Li

saurabhvyas commented 5 years ago

I am also interested in doing this, testing it with ocr datasets, and see how it works