search

Wizaron / instance-segmentation-pytorch

Semantic Instance Segmentation with a Discriminative Loss Function in PyTorch
https://arxiv.org/abs/1708.02551
GNU General Public License v3.0
458 stars 95 forks source link

Instance Counter #15

Closed JadTawil-theonly closed 4 years ago

JadTawil-theonly commented 4 years ago

Hello Sir,

Why is the last layer in the instance counter a sigmoid layer?

Would this not squash the output to the range 0->1.

This does not seem to make sense if you want this to count the number of instances, should the output not be the Linear layer?

JadTawil-theonly commented 4 years ago

Nevermind! I see now that the number of instances is normalized to 0->1 before being used as ground truth, so the sigmoid actually makes sense.