copySplitImage

最新推荐文章于 2024-09-10 13:01:50 发布

ONLY9598

最新推荐文章于 2024-09-10 13:01:50 发布

阅读量81

点赞数

分类专栏： # python 文章标签： python 开发语言

本文链接：https://blog.csdn.net/qq_34508123/article/details/129288311

版权

python 专栏收录该内容

2 篇文章 0 订阅

订阅专栏

python整理文件夹数据

记录用

我的文件夹下存在两种格式的数据（jpg, png）。手动划分太慢了，写了一个python代码进行划分。划分前的文件夹如下：

在这里插入图片描述

import os
import time
from tqdm import tqdm
import shutil
import argparse

# 
parser = argparse.ArgumentParser()
parser.add_argument('--data_path', default='../DUTS-TE/', type=str, help='source image path')
parser.add_argument('--image_save_path', default='../dutstestimage/', type=str, help='The storage path of the divided image')
parser.add_argument('--label_save_path', default='../dutslabelimage/', type=str, help='The storage path of the divided label')
args = parser.parse_args()

class splitFile():
    def __init__(self, args) -> None:
        super().__init__()

        self.makeDir(args.image_save_path)
        self.makeDir(args.label_save_path )

        self.copyImage(args.data_path, args.image_save_path, args.label_save_path)

    def makeDir(self, path):
        if not os.path.exists(path):
            os.mkdir(path)

    def copyImage(self, source_path, image_path, label_path):
        imageList = tqdm(os.listdir(source_path), colour="red")
        for f in imageList:
            time.sleep(0.01)
            if f.endswith("png"):
                shutil.copy(source_path + f, label_path + f)
            elif f.endswith("jpg"):
                shutil.copy(source_path + f, image_path + f)
            
            imageList.set_description("Processing %s" % f)


if __name__ == "__main__":
    splitFile(args)