Classification Layer - Activation & Shape?
stalhabukhari opened this issue · comments
Talha Bukhari commented
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 commented
have you run the code ,is it working? giving desirable result?
sneh-debug commented
in old implementation you can see softmax activation have been used.