Open June-Jo opened 3 years ago
I was confused of indexing method in data loader. I noticed that 0 is not background or unlabeled but the road..... Therefore, I changed ignore_index from 0 to 250 back.
Now the mIoU is 52% after 4 epochs and it looks like keep increasing.
I'll report after the training finishes.
P.S. I added more augmentations like below.
rotate: 10
rsize: 1024
hflip: 0.5
vflip: 0.5
brightness: 0.5
saturation: 0.5
contrast: 0.5
gamma: 0.5
hue: 0.5
pspnet reach at 65% of mIoU. Any tips to increase further?
Hi, I'm trying to train the segmentation models using cityscape dataset(training: 2975, validation:500).
In the papers of some segmentation models, mIoU of them is over 70%. However, I cannot reach there.
I tried pspnet, segnet, and fcns, but they just reached to 40% of mIoU. I also tried data augmentation(scaling, random rotation, horizontal flip). In order to ignore the background, I changed the ignore_index of loss function from 250 to 0, and the label of 250 is replaced to 0.
Did I do something wrong or is there any tip to train them from scratch?
Thanks for your help in advance.
My cfg file of pspnet is below.