dataset-division-to-train-val-test-python
python script for dividing image data to train test and val
If you have an image dataset in the following structure of directories
- dataset
- label_1 directory
- label_2 directory
- label_3 directory
- label_n-1 directory
- label_n directory
then this python script (main.py) will divide the images present in each label directories into train, val and test directories.
set the following variables according to your requirements in main.py
PATH_OF_DATA= './dataset/'
train_perc = 0.7 #70%
val_perc = 0.2 #20%
test_perc = 0.1 #10%