diff --git a/copy_randomic_files.py b/copy_randomic_files.py index 7429c88..41231e9 100644 --- a/copy_randomic_files.py +++ b/copy_randomic_files.py @@ -2,7 +2,7 @@ import shutil import random -from settings.settings import COPY_SOURCE_FOLDER, DATA_PATH, COPY_FIXED_NUMBER_TO_COPY, TRAIN_PERCENTAGE +from settings.settings import COPY_SOURCE_FOLDER, DATASET_FOLDER, COPY_FIXED_NUMBER_TO_COPY, TRAIN_PERCENTAGE def copy_files( @@ -43,7 +43,7 @@ def copy_files( if __name__ == '__main__': copy_files( COPY_SOURCE_FOLDER, - DATA_PATH, + DATASET_FOLDER, fixed_number=COPY_FIXED_NUMBER_TO_COPY, random_mode=True ) diff --git a/settings/settings.example.py b/settings/settings.example.py index eefc057..86a2d0b 100644 --- a/settings/settings.example.py +++ b/settings/settings.example.py @@ -6,6 +6,7 @@ # COPY RANDOMIC FILES COPY_SOURCE_FOLDER = '__source_images_folder_path__' +DATASET_FOLDER = './dataset' TRAIN_PERCENTAGE = 0.7 COPY_FIXED_NUMBER_TO_COPY = 32