文章目录
前言
最近在学习神经网络,需要划分训练样本,我想将训练图像和标签的命名保持一致,这就需要将总的目录下相同文件名的图像复制(剪切)到新的文件夹。
以下是本篇文章正文内容,下面案例可供参考
一、学习这篇博客,你将获得
将指定后缀名或文件中具有相同字符的文件复制(剪切)到新的文件夹。
二、代码
代码如下(示例):本文将带有.jpg和.JPG后缀名的图像复制(剪切)到新的文件夹
''' author: id:dream_me 备注:这是完整代码,可直接运行 编译环境:python3.6 '''
import os
import cv2
import numpy as np
from PIL import Image
import matplotlib.pyplot as plt
import shutil
new_image_folder = r'新的文件路径'
dir_path = r'存放所有图像的路径'
#遍历读取文件夹筛选符合标准的图片
for root, dirs, files in os.walk(dir_path):
# root 表示当前正在访问的文件夹路径
# dirs 表示该文件夹下的子目录名list
# files 表示该文件夹下的文件list
for file in files:
if file.count('.jpg') or file.count('.JPG') == 1:
shutil.copy(os.path.join(root,file),new_image_folder) #复制图片到新文件夹
os.remove(os.path.join(root,file)) #删除名称带有'.jpg'和'.JPG'的图像
总结
如果只需要实现复制功能,请去掉代码最后一行。
以上就是今天要讲的内容,本文简单地实现了批量将指定图像复制(剪切)到新的文件夹下。