Data set preparation and arrangement

coding: utf-8

In[3]:

import os
# For copying files
import shutil

Create a new folder to store the sorted pictures

for i in range(17):

# 17 Categories NEW 17 A folder 0-16

os.mkdir(‘D:/17flowers’+’/’+‘flower’+str(i))

Cycle through the pictures of all the flowers

#os.listdir() Method to return a list of the names of the files or folders contained in the specified folder .
for i,path in enumerate(os.listdir(‘D:/17flowers/jpg/’)):
# Define the full path of the flower picture
image_path = ‘D:/17flowers/jpg/’ + path
# Copy to the corresponding category , Each category 80 A picture
#copyfile(src, dst)： take src Copy the contents of the file to dst file
shutil.copyfile(image_path, ‘D:/17flowers’+’/’+‘flower’+str(i//80)+’/’+path)