批量移动图片/文件(附python代码)

因为我的文件夹结构是
fourthdown/CVPR/ababab/train_sets

   ——1
   	  ——1
  	  ——2
	  ……
   	  ——n
   ——2
   ……
   ——n

然后该代码的功能是,移动图片,使得每七张图片存在一个新的文件夹中,移动时,如果被处理的文件夹中当前图像/文件有后六个文件,才进行移动,否则进入下一个文件夹。

举个例子,一个文件夹中有10个图片/文件,那么1-7,2-8,3-9,4-10,这四组各七张图片/文件存入新的文件夹中,命名为1、2、3、4,然后该文件夹处理完毕,继续处理下一个文件夹。

import os
import shutil

def moveFile(fileDir,tarDir):
    if not os.path.isdir(fileDir): #查看数据集是否存在
        print('Error: No source data found')
        exit(0)
    if not os.path.isdir(tarDir): #查看存储文件夹是否存在,不存在则创建,train
        os.mkdir(tarDir)

    paths = os.listdir(fileDir)
    for path in paths:
        source_dir = fileDir + '/' + path #H:/fourthdown/CVPR/ababab/train_sets/1
        path_1 = tarDir + '/' + path #H:/fourthdown/CVPR/ababab/train/1
        if not os.path.isdir(os.path.join(tarDir, path)): #/train/1
            os.mkdir(os.path.join(tarDir, path))

        count = 1
        dirs = os.listdir(source_dir)
        for dir in dirs:
            dir_path = source_dir + '/' +dir  #H:/fourthdown/CVPR/ababab/train_sets/1/1

            file_list = os.listdir(dir_path)
            num_classes = len(file_list) #/1/1图片数目

            for i in range(num_classes): #遍历/1/1所有图片
                if ((i+6) < num_classes): #当前图片有后七张图片
                    path_2 = path_1 + '/' + str(count)  # H:/fourthdown/CVPR/ababab/train/1/1
                    if not os.path.isdir(path_2):
                        os.mkdir(path_2)

                    count = int(count) + 1

                    for n in range(1,7):
                        class_name = file_list[i]
                        save_name = 'im' + str(n) + '.png'
                        image_dir = dir_path + '/' + class_name
                        path_3 = path_2 + '/' + save_name
                        shutil.copyfile(image_dir,path_3)

if __name__=='__main__':
    fileDir = "H:/fourthdown/CVPR/ababab/test_sets" #原始图片文件夹路径
    tarDir = "H:/fourthdown/CVPR/ababab/test" #移动到新文件夹的路径
    moveFile(fileDir,tarDir)
  • 0
    点赞
  • 4
    收藏
    觉得还不错? 一键收藏
  • 0
    评论

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值