Eyal:

I understand that this affects the training process, so we should initialize the weights with respect to the proportion of the classes.


I don’t think so this  a good idea, a better idea would be ground plane augmentation.