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)