NIRVANALAN / PyTorch_UNOdeMSegNet

UNOdeMSegNet for Medical segmentation
2 stars 4 forks source link

Microscopy segmentation exps #1

Open NIRVANALAN opened 5 years ago

NIRVANALAN commented 5 years ago

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:

settings best ACC best loss
1000 unfiltered 93.157772 0.292162
800 unfiltered
1000 unfiltered, lr=0.02 93.151519 0.300175
NIRVANALAN commented 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.

  1. Pixel Accuracy image
  2. Multi Class Pixel Accuracy image
  3. Multi Class IOU image