search

rezazad68 / BCDU-Net

BCDU-Net : Medical Image Segmentation
712 stars 264 forks source link

Semantic bug in the implementation #37

Closed mmheydari97 closed 2 years ago

mmheydari97 commented 2 years ago

I believe using go_backwards=True does not make a ConvLSTM bidirectional. It just makes the input reversed. Instead, there should be a Bidirectional layer wrapping around the ConvLSTM to do the job. This way there would be two separate outputs for forward and backward cells. I'm not sure if those should be concatenated or averaged based on the model proposed in the paper. For the first option we have to make the number output channels half, so that the dimensions stay constant.

rezazad68 commented 2 years ago

Thanks for your interest in our work. This issue was addressed in our closed issue before. Please check #8

Best regards