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.

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