Open NIRVANALAN opened 5 years ago
Considering the BG pixels (pixels that are not labeled) dominates the images, category CE loss may lead to overfitting(all pixels are predicted as BG pixels, and the ACC is still high), this time I tried with BCE+DICE Loss and trained models without BG pixels. The output of different class lies in different channel, and softmax as final layer activation function.
Also, 3 metrics are used to evaluate the model.
All data patches data split:
Trainset: 1.8k Testset: 400
Remove pure BG images(without organelles)
Train: 1444 test: 362
results on several dataset split: