1

I am doing a brain MRI segmentation task using Unet model. My problem is it does no matter what transformation i use during training my val dice score after few epochs stucks around 77% score.

I already have a successfuly trained model with 93% dice score on my test set, but i wanted to improve it by adding some augmentation to my training. I used transformations from the albumentation library. I only experimented with transformations that are reveleant to my train case. (Horizontal, VerticalFlip, etc ..) Things that i already tried:

  • waiting for more epochs
  • using different transformations, using only one
  • using different encoders to make the model more deep, (resnet34, resnet152 etc)

0 Answers0