search

ThienAnh / PSPNet

0 stars 0 forks source link

How to fix the class imbalance in PSPNet? #2

Open bhadresh74 opened 6 years ago

bhadresh74 commented 6 years ago

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 commented 6 years ago

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).