| 1234567891011121314151617181920212223 |
- def populate_kfold_directories(data_dir, K_FOLDS):
- alarmed_images = os.listdir(f"{data_dir}/Alarmed")
- annoyed_images = os.listdir(f"{data_dir}/Annoyed")
- curious_images = os.listdir(f"{data_dir}/Curious")
- relaxed_images = os.listdir(f"{data_dir}/Relaxed")
- for i in range(K_FOLDS):
- validation_range = (i*20, i*20 + 20)
- for j in range(0, 100):
- if validation_range[0] <= j < validation_range[1]:
- shutil.copy(f"{data_dir}/Alarmed/{alarmed_images[j]}", f"folds/fold{i}/validation/Alarmed/")
- shutil.copy(f"{data_dir}/Annoyed/{annoyed_images[j]}", f"folds/fold{i}/validation/Annoyed/")
- shutil.copy(f"{data_dir}/Curious/{curious_images[j]}", f"folds/fold{i}/validation/Curious/")
- shutil.copy(f"{data_dir}/Relaxed/{relaxed_images[j]}", f"folds/fold{i}/validation/Relaxed/")
- else:
- shutil.copy(f"{data_dir}/Alarmed/{alarmed_images[j]}", f"folds/fold{i}/train/Alarmed/")
- shutil.copy(f"{data_dir}/Annoyed/{annoyed_images[j]}", f"folds/fold{i}/train/Annoyed/")
- shutil.copy(f"{data_dir}/Curious/{curious_images[j]}", f"folds/fold{i}/train/Curious/")
- shutil.copy(f"{data_dir}/Relaxed/{relaxed_images[j]}", f"folds/fold{i}/train/Relaxed/")
|
