.venv
__pycache__

augmented_dataset

training