python_for_microscopists icon indicating copy to clipboard operation
python_for_microscopists copied to clipboard

need advise

Open IkhlasAhmad1998 opened this issue 4 years ago • 0 comments

Hello Dr, i tried to contact you on linkedin but seems like you aren't there. First of all thanks for such a great repo that surely helps alot. BTW i am doing semantic segmentation on a small dataset of endoscopic images. i am able to train a U-NET using segmentation models. i have an image like this: image and respective mask for 5-classes excluding background: image

i have increased the number of training images and masks using augmentation but still not getting good results. i have training iou of 0.53 while testing iou is just 0.45. i am using sigmoid activation function as i needed multi-output ( there is overlap in masks)

  1. please tell me how to improve my IOU in this case.
  2. can i use same augmentation (that i used to increase dataset) during training to overcome overfitting.
  3. how can i deal this problem as One-vs-REST.

waiting for your kind response Thanks :)

IkhlasAhmad1998 avatar Sep 14 '21 10:09 IkhlasAhmad1998