python:从文件夹中挑选图片,复制到另一文件夹并重命名

python:从文件夹中挑选图片,复制到另一文件夹并重命名

  • 前言
  • 一、功能描述
  • 二、具体实现
  • 总结


前言

应用于目前的项目,在处理图像数据集的时候,经常需要将原始图像数据中的某些图像进行整理,便于后续训练分类模型制作标签。代码参考了数篇优秀之作,过程中发现很多时候能够找到的解决类似问题的代码往往有着版本、环境不适配的问题,进行一定修改才能真正解决自己的需求。这里很多放上来的代码也都是大家修改成解决某个具体问题的版本,所以不要想着偷懒,不要以为能够找到直接能用一点都不需要改的代码——提醒自己。

一、功能描述

一段用来挑选图片的python代码:

  1. 将图片从文件夹A复制到文件夹B
  2. 图片在复制后进行重命名
  3. 自定义图片的新命名格式

二、具体实现

代码如下:

import os
import shutil

path = 'E:\\move\\桌面\\2'
new_path = 'E:\\move\\桌面\\1'
filelist = os.listdir(path)
i = 1

for item in filelist:
    if item.endswith('5.bmp') or item.endswith('6.bmp') or item.endswith('7.bmp') or item.endswith('8.bmp'):
        src = os.path.join(os.path.abspath(path), item)
        dst = os.path.join(os.path.abspath(new_path),item)
        new_name = os.path.join(os.path.abspath(new_path),''+str(i)+'.bmp')
       #复制图像
        shutil.copy(src,dst)
       #重命名
        os.rename(dst, new_name)
        i += 1

        print(src)
        print(new_name)

这里选出的是后缀为“xx5.bmp”、“xx6.bmp”、“xx7.bmp”、“xx8.bmp”的图片,if条件语句结合“or”并列多个筛选条件,下图为运行结果:
python:从文件夹中挑选图片,复制到另一文件夹并重命名_第1张图片


总结

shutil包
os包

参考博文:
python实现将子文件夹下的图片复制到新的文件夹并重命名
基于Python 从一个文件夹读取照片,修改尺寸后重命名存入另一个文件夹

你可能感兴趣的:(python应用,python)