套娃式文件夹如何通过Python批量处理_Python

套娃式文件夹如何通过Python批量处理

2020-08-24 00:03EEEEEdan Python

这篇文章主要介绍了套娃式文件夹如何通过Python批量处理，文中通过示例代码介绍的非常详细，对大家的学习或者工作具有一定的参考学习价值，需要的朋友们下面随着小编来一起学习学习吧

前言

在我对项目组的一些训练图像进行预处理的时候，发现处理的图像是分好了类，在文件夹里的文件夹里，套娃式存储的，所以对我批处理，以及按原文件夹规则进行存储的时候，就会造成很大困扰
但通过下面几个函数的结合，帮我顺利的完成了一系列的预处理。

一、用不腻的芷山库

1.安装库

pip安装：

1	`pip install zisan`

2.getFiles函数

函数调用：

									import zisan.FileTools as zf

									file_path = 'C:/Users/xxx/Desktop/2016/Annotations'

									whole_file = zf.getFiles(file_path) 

图片存储在：件夹2016 -> 文件夹Annotations ->子文件夹 -> 00000.png
通过getFiles函数，可以将Annotations中的所有文件夹里的所有图片路径调出来

也就是说，getFiles函数是将文件夹里面的所有文件路径调出来，不管中间有无子文件夹

二、其他函数

1.os.listdir函数

该函数调用后会返回路径底下文件夹的名称，以字符串的形式储存在列表里

代码如下：

									import os

									file_path = 'C:/Users/xxx/Desktop/2016/Annotations'

									file_names = os.listdir(file_path)

									print(file_names)

效果：

套娃式文件夹如何通过Python批量处理

2.os.mkdir函数

代码：

									import os

									new_file_path = 'C:/Users/xxx/Destop/2016/newfile'

									os.mkdir(new_file_path)

用于创建新文件夹

三、运用

要求：处理Annotations文件夹中的每个子文件夹中的图片，并按原本的规则存储在newfile新文件夹里的相应位置，命名规则如00000.jpg

									import zisan.FileTools as zf

									import os

									import cv2

									from skimage import io

									file_path = 'C:/Users/xxx/Desktop/2016/Annotations'

									new_file_path = 'C:/Users/xxx/Destop/2016/newfile'

									file_names = os.listdir(file_path)

									#获取Annotations文件夹的子文件夹名称

									for i in file_names: #遍历每个子文件夹名称

									  Index = 0

									  file_name = file_path + '/' + i #巧妙运用+号得到改子文件夹的路径

									  os.mkdir(new_file_path + '/' + i) #在newfile里创建一个与子文件夹名称相同的文件夹

									  whole_pic = zf.getFiles(file_name) #用getFiles函数读取子文件夹内的图片路径

									  for f in whole_pic:

									    msk = io.imread(f)

									        msk=cv2.cvtColor(msk,cv2.COLOR_RGBA2GRAY)

									    msk[np.where(msk!=0)]=255

									    io.imsave(new_file_path + '/' + i + '/' + str("%05d" % Index) + '.jpg' , msk)

									     #处理命名可直接+'.jpg'让其以jepg形式存储

									    Index += 1