我是为了提高图片浏览效率,批量整理文件夹
将图像样本数多于2000个的文件夹按 2000个图像样本/一个子文件夹 存放
#子文件夹命名规则为01,02,03…
import os
import shutil
import glob
if __name__ == '__main__':
# 需整理的文件夹
srcpath = input("请输入需要整理的文件夹路径:" + "\n")
for dirpath, dirname, filename in os.walk(srcpath):
# 子文件夹数,子文件夹命名为‘01’,‘02’...
n = 1
for file in filename:
# 子文件夹
dstpath = srcpath + "//" + '0' + str(n)
# 图像样本计数
filenum = len(glob.glob(dstpath + "/*"))
while os.path.exists(dstpath) and filenum >= 2000:
n += 1
dstpath = srcpath + "//" + '0' + str(n)
filenum = len(glob.glob(dstpath + "/*"))
#待整理样本文件数
num = (len(glob.glob(srcpath + "/*"))) - n
if num == 0:
print("文件整理结束")
quit()
if not os.path.exists(dstpath):
os.makedirs(dstpath)
src_file = os.path.join(srcpath, file)
shutil.move(src_file, dstpath)