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开头的文件全被移到目标路径下了。

至此，本博文就结束了。如果该博文对你有所帮助的话，麻烦点赞、收藏，方便下次快速地找到，从而提高效率。