mahmoudnafifi / mixedillWB

Reference code for the paper Auto White-Balance Correction for Mixed-Illuminant Scenes.
Other
88 stars 17 forks source link

Gridnet diagram in readme is mismatching from code #14

Open a-shivgan opened 5 days ago

a-shivgan commented 5 days ago

image There is incorrect representation of number of channels in the residual block expansion's second convolution layer. The number of input channels to the second convolution block should be '2c' and the number of output channels should be 'c'. In the code, it is maintaining the constant number of channels horizontally, but the same is not reflected in the diagram.