pytorch-CycleGAN-and-pix2pix icon indicating copy to clipboard operation
pytorch-CycleGAN-and-pix2pix copied to clipboard

Training with data that has more than 3 channels

Open ne1114 opened this issue 3 years ago • 2 comments

Hello I'm trying to train with data that are more than 3 channels (ex 6, 12).

After the training I noticed that the model hasn't learned sufficiently compared to 3 channel data. (The 3, 6, 12 channeled datasets have images of interest stacked channel wise. Which means its basically grayscale image stacked 3 times vs. 6 times. 12 times.)

I think it would be hard for the model fully capture the importance of channel wise just with L1 loss, as I stack more channels.

Would there be any suggestions other than changing the loss function? For example, increase the ngf, ndf, stack more network layers, etc please?

ne1114 avatar Nov 26 '21 05:11 ne1114

I am not sure if changing hyper-parameters will help your case. Depending on the type of data you are working on, you might need a new reconstruction loss rather than the default L1 loss.

junyanz avatar Dec 02 '21 20:12 junyanz

Hello, I would like to ask you how did you implement the training of more than 3 channels of data, and which part of the code was modified?

tongfh1 avatar Aug 14 '23 01:08 tongfh1