I was trying one of the CNN tutorials for shape Image classification using python. I have 4 folders of different shapes. I try the following code to split the dataset.