costapt / vess2ret

A Keras implementation of pix2pix code adapted to generate retinal images from vessel networks.
MIT License
94 stars 42 forks source link

merge() in concatenate_layers() is being called with mode='sum' when using Keras 1 #4

Closed aitorshuffle closed 7 years ago

aitorshuffle commented 7 years ago

Hi,

Since the addition of the compatibility with Keras 2, the call to merge() in concatenate_layers() no longer includes the mode parameter when used with Keras 1. Thus, merge() is now called with the default mode='sum' default value instead of with 'concat', which I understand that is not the expected behaviour (and crashes if the number of channels is different for A and B).

Many thanks for sharing this work!

Best,

Aitor

costapt commented 7 years ago

Hi,

Thank you for pointing out the error. The merge layer is now using the mode parameter and so I think this issue is solved now.

Thanks again.