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 ( will divide the images present in each label directories into train, val and test directories.

set the following variables according to your requirements in

PATH_OF_DATA= './dataset/'
train_perc = 0.7 #70%
val_perc = 0.2 #20%
test_perc = 0.1 #10%


View Github