Classification Layer - Activation & Shape?

[stalhabukhari]

Hi!

I went through this repository after reading your paper. Architecture on page 6, shows the final classification layer to produce feature maps of shape (240, 240, 2) which may indicate the use of a Softmax activation (not specified in the paper).
On the contrary, model used in code has a classification layer of shape (240, 240, 1) using Sigmoid activation.

Kindly clarify this ambiguity.

[sneh-debug]

have you run the code ,is it working? giving desirable result?

[sneh-debug]

in old implementation you can see softmax activation have been used.