Modify these variables to your dataset/output path
project_dir = r'YOUR_PROJECT_DIR'
task_name = 'YOUR_DATASET_NAME'
data_path = os.path.join(project_dir,task_name,'data.yaml') # PATH for yaml
output_folder = r'/content/drive/MyDrive/Colab Data/output/{}'.format(task_name)