前言
虽然有时候手动复制图片到另外一个文件夹非常方便,但是在制作深度学习数据集有时候还是需要用代码进行统一批量处理,今天就分享一个自己写的简单的复制图片的小程序。一、使用步骤
1.引入库
代码如下:
import os #用于对文件惊醒操作
import shutil #在这里只用到了将图像从源路径复制到指定路径
2.代码
代码如下:
import os
import shutil
def read_name():
source_file_path0 = 'E:\\数据集\\guofeng\\' #注意opencv库不能打开中文路径
name0 = os.listdir(source_file_path0) #读取源路径下所有文件的名称,将其放在一个列表内返回,每个元素代表一个文件名
new_file_path0 = 'E:\\数据集\\guofeng1\\'
return name0, new_file_path0, source_file_path0
if __name__ == "__main__":
name, new_file_path, source_file_path = read_name()
print(name)
for i in name:
if 'b' in i and '45d' in i: #对特定字符进行判断,若是有这几个字符,就复制过去
middle_file_name = source_file_path + i #指定文件的路径为路径名加文件名
shutil.copy(middle_file_name, new_file_path) #左边是源文件的路径加文件名称,后面是目标路径,可不加文件名称