批量从多个文件夹中提取图片的小脚本

import os
import shutil

path = 'F://maskrcnn tang//MASK-RCNN//mask rcnn_test//mydata//labelme_json//'  # _json文件夹所在的路径
new_path = 'F:\\maskrcnn tang\\MASK-RCNN\\mask rcnn_test\\mydata\\cv2_mask'  # 需保存的路径
count = os.listdir(path)
for j in range(0, len(count) + 1):
    for root, dirs, files in os.walk(path):
        if len(dirs) == 0:
            for i in range(len(files)):
                print("i=", i)
                if files[i].find('label.png') != -1:
                    # zfill(3)为数值的长度，如0001_json,长度是4
                    shutil.copy(os.path.join(path + '/' + str(j).zfill(0) + '_json/', files[i]),
                                os.path.join(new_path, str(j).zfill(0) + '.png'))

后续改进：由于数据集处理需要，后续对程序有了符合需求的改进：效果是：从单一文件夹中提取某后缀名为.tif的文件。如果有报错，请与我联系。

import os
import shutil

path = 'G://maskrcnn tang//data enhancement//tools-master1//tools-master//DataAugForObjectSegmentation//data3/'  # _json文件夹所在的路径
new_path = 'G:\\maskrcnn tang\\data enhancement\\tools-master1\\tools-master\\DataAugForObjectSegmentation\\data4'  # 需保存的路径
count = os.listdir(path)
for j in range(51, len(count) + 1):
    for root, dirs, files in os.walk(path):
        if len(dirs) == 0:
            for i in range(len(files)):
                print("i=", i)
                if files[i].find('.tif') != -1:
                    # zfill(3)为数值的长度，如0001_json,长度是4
                    # shutil.copy(
                    #             os.path.join(new_path, str(j).zfill(0) + '.tif'))
                    # shutil.copy(os.path.join(path + '/' + str(j).zfill(0) + '_json/', files[i]),
                    #             os.path.join(new_path, str(j).zfill(0) + '.png'))
                    # print(str(j).zfill(3))
                    shutil.copy(os.path.join(path + '/' +"Image_"+ str(j).zfill(3)+".tif"),
                                os.path.join(new_path, 'Image_'+str(j).zfill(3) + '.tif'))

感谢博主

python实现将子文件夹下的图片复制到新的文件夹并重命名_蹦跶的小羊羔的博客-CSDN博客_python复制图片并重命名