search

seoungwugoh / RGMP

Fast Video Object Segmentation by Reference-Guided Mask Propagation
192 stars 37 forks source link

Why use softmax instead of sigmoid? #17

Open evbtst opened 4 years ago

evbtst commented 4 years ago

line 66: msv_E2[sc] = upsample(F.softmax(e2[0], dim=1)[:,1].data.cpu(), (h,w))

line 71: msv_E2[sc] = F.softmax(e2[0], dim=1)[:,1].data.cpu()

If you run these lines for each object (Propagate_MS is inside the loop for o in range(num_objects) ) , why use softmax instead of sigmoid?

seoungwugoh commented 4 years ago

Hi @evbtst, We perform 2-way (FG & BG) softmax for the binary segmentation. The similar operation can be implemented using the sigmoid. But, we simply choose to use the softmax.