search

junyanz / CycleGAN

Software that can generate photos from paintings, turn horses into zebras, perform style transfer, and more.
Other
12.34k stars 1.94k forks source link

Question about the input in cyclegan #107

Closed amy-hyunji closed 5 years ago

amy-hyunji commented 5 years ago

Hi I want to ask question about the input channels when training cyclegan. I tried to have input channel as 'L':1 channel and output as 'RGB':3 channels. However, there was an error message saying that there is trouble on number of channels. I thought that this might be because the input and output should have same number of channels. Is my prediction right or is it due to somewhere else in code?

Thank you :)

junyanz commented 5 years ago

I think you need to set input_nc 1 and output_nc 3. Also, check out our latest PyTorch code with colorization mode.