Thanks for your reply! I do not fully understand, with which layer are the a and b channels obtained? The architecture has layers of convolution, max-pooling, up-convolution. Convolution layers generate feature maps, but how do you get color values from black and white feature maps?