I have an imbalanced dataset and I want to generate more images from one type and train on it as well. However, it seems like that the different statistics of the generated images differ from the original dataset. Any idea how can I align them?