Class Hierarchy

src.dataset_generator.DatasetGenerator - Generate dataset for training.
src.openai_translator.CaptionTranslator - Translate the image captions to English via OpenAI API.
src.statistics.Statistics - Calculate the statistics of the dataset.