Class imbalance in one hot encoding for CNN
You might want to consider using metrics such as the f1-score in order to measure how well your model does.

Notice that if, in real life, the inbalance will be of a similar nature, then your model has to account for it anyway.