keras-unet-collection icon indicating copy to clipboard operation
keras-unet-collection copied to clipboard

Possible bug in number of tensors for layer fusion

Open TheOnlyError opened this issue 1 year ago • 1 comments

The figure below shows that the value of a decoder neuron is based on 5 other values:

image

When trying to reproduce this in the code, I get the following:

image

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?

TheOnlyError avatar Jul 31 '22 13:07 TheOnlyError

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.

ziyangwang007 avatar Sep 16 '22 00:09 ziyangwang007