Python批量删除或移动指定图像
前言
笔者的研究方向为计算机视觉,因此经常和大量图像打交道,有时需要批量删除一些图像,有时需要批量移动一些图像,因此编写了下述代码。下述代码可以实现 删除指定名称的图像 或 移动指定名称的图像 。
一、批量删除指定名称的图像
删除前,路径【D:\basic\aligned】下包括如下图像,既有命名以test开头的图像,也有以train开头的图像。
下述代码实现了删除指定路径,即【D:\basic\aligned】下所有命名以test开头的图像。
# 批量删除指定名称的图像
root = r'D:\basic\aligned'
for file in os.listdir(root):
if file.startswith('test'): # 删除root路径下 命名以'test'开头的图像
os.remove(os.path.join(root, file))
代码效果如下,可以看到,所有命名以test开头的图像都被删除了。
二、批量移动指定名称的图像
移动前,原路径,即【D:\basic\aligned】下包括如下图像,既有命名以test开头的图像,也有以train开头的图像。
目标路径,即【D:\compound\aligned】下什么图像也没有
下述代码实现将原路径【D:\basic\aligned】中所有命名以train开头的图像移动到目标路径【D:\compound\aligned】。
src = r'D:\basic\aligned' # 原文件夹
dst = r'D:\compound\aligned' # 目标文件夹
for file in os.listdir(src):
if file.startswith('train'): # 移动原文件夹中 命名以test开头的图像 到目标文件夹
shutil.move(os.path.join(src, file),os.path.join(dst, file))
移动之后可以看到,原路径下只剩下命名以test开头的图像,所有以train开头的文件全被移到目标路径下了。
至此,本博文就结束了。如果该博文对你有所帮助的话,麻烦点赞、收藏,方便下次快速地找到,从而提高效率。