[python]将不同类别文件夹下面的图像搬到同一文件夹下

# -*- coding: utf-8 -*-# 用于将不同类别的文件夹下的图片搬到某个文件夹，并在文件前面加上目录名字:#比如caltech256，里面有256个文件夹，每个文件夹放的都是一类，用movefiles.py可以生成一个database的文件夹，#这个文件夹把databaseClassified的图片都搬到里面来，并且生成待查询的图片queryImgs.txt和databaseClasses.txt# python movefiles.pyimport osimport fnmatchimport shutilimport randomimport mathquery_number_percent = 0.5 # 设置每类拿百分之多少出来作为查询directory = "database"  # 设置新路径databaseClasses = 'UCF-101'if not os.path.exists(directory):    os.makedirs(directory)newImgDBPath = os.path.abspath(directory)# walk through the folderf = open("./databaseClasses.txt", "w")g = open("./queryImgs.txt", "w")for root, dirs, files in os.walk(databaseClasses):    for i, str_each_folder in enumerate(dirs):        # we get the directory path        str_the_path = '/'.join([root, str_each_folder])        files_number = len((os.listdir(str_the_path))) #子目录下文件数目        # 生成查询图片实例        index = random.sample(range(0, files_number), int(math.floor(query_number_percent*files_number)))        # list all the files using directory path        for ind, str_each_file in enumerate(os.listdir(str_the_path)):            # now add the new one            str_new_name = '{0:03}'.format(i+1) +'_'+ str_each_folder + '_' + str_each_file            if ind in index:                g.writelines('%s\n' % str_new_name)                # full path for both files            str_old_name = '/'.join([str_the_path, str_each_file])            str_new_name = '/'.join([newImgDBPath, str_new_name])            # now rename using the two above strings and the full path to the files            # os.rename(str_old_name, str_new_name) # 搬运原文件到设置的新目录下            shutil.copy2(str_old_name, str_new_name)  # 拷贝原文件到设置的新目录下        #  we can print the folder name so we know that all files in the folder are done        print '%s, %d images' % (str_each_folder, files_number)        f.writelines('%s %d\n' % ('{0:03}'.format(i+1)+'_'+str_each_folder, files_number))g.closef.close