keras-unet-collection
keras-unet-collection copied to clipboard
Possible bug in number of tensors for layer fusion
The figure below shows that the value of a decoder neuron is based on 5 other values:
When trying to reproduce this in the code, I get the following:
The number of tensors is equal to 4 instead of 5. This suggest there is a bug in the code around the following lines:
https://github.com/yingkaisha/keras-unet-collection/blob/d30f14a259656d2f26ea11ed978255d6a7d0ce37/keras_unet_collection/_model_unet_3plus_2d.py#L142-L150
I think the second for loop should have depth_decode+1
as the upper boundary in the range. Is this indeed a bug?
I also noticed that the X1EN can not be copied and crop to the decoder part through multi-scale skip connections each time. I agree with you.