Saeed Izadi
Saeed Izadi
Hi, Is there any fix/update on this?
I've modified the models by @bodokaiser for a binary segmentation problems on medical images (using nn.Sigmoid + BCELoss) and hence my pretrained models may not come helpful in general. However,...
Hi @ZweeLe Yes, I wanna share the modifications but I'm a bit out of time these days. You must change the num_classes to 1 instead of 2. Then, simply change...
You would suggest to use Sigmoid() + BinaryCrossEntropy for Binary Segmentation.
@woaichipinngguo You should see the definitions for the parameters. But, for the initial try, just try to use the default values, and see the results, then, you can customise them...
Hi, That's so weird! so so weird!! Although there are some problems with your UNet implementation, but I found that the blank issue does not arise from UNet, or generally...
I finally found the problem!! For the last set of convolutions, that is 128-> 64 -> 64 -> 1, the activation function should not be used! The activation function causes...
Sure, I will do a pull request
@bodokaiser thank you for your information :)
Hi JiayuanSternLi By blank image, I meant an image in which all pixel labels are 0 or very small values, e.g. 0.01 in the range of [0,1].