search

irfanICMLL / TorchDistiller

188 stars 24 forks source link

Binary image segmentation #3

Open rose-jinyang opened 2 years ago

rose-jinyang commented 2 years ago

Hello How are you? Thanks for contributing to this project. I have a question. As in the previous paper & project, u handled ONLY the multi-class image segmentation problem. I have some trouble using your KD methods in my project for binary image segmentation I used sigmoid rather than softmax because the output has 1 channel rather than 2 channels.

image

But this implementation does NOT play its role. Can the conventional KD or your CWD method be used for binary image segmentation? Thanks

luowei0701 commented 2 years ago

Have you successfully applied binary img seg?