How to fix the class imbalance in PSPNet?

[bhadresh74]

Hi,

I followed the same repo https://github.com/hszhao/PSPNet for training the PSPNet.
Since I have class imbalance, I get the per class accuracy 0 for all non-dominant classes.
If you have any solution, could you please share with me?

[ThienAnh]

Are you try reduce learning rate? I has set lr = 0.000005, and set crop size = 377 and that seem success.
This is result with VOC2012.

https://www.dropbox.com/sh/9dllcdybaqvseoz/AAAT-wQZA-ySYyuSQqC9zid1a?dl=0
I'm finding solution for train my data :(. ( I don't understand program can be train on VOC data with SegmentationClassAug (Has only contour image in this folder, how mapping with color map).