【数据集处理】Python 将图片名字中含有某一字符串的图片存到指定文件夹下 |
您所在的位置：网站首页 › yl字母的图片 › 【数据集处理】Python 将图片名字中含有某一字符串的图片存到指定文件夹下 |
【数据集处理】Python 将图片名字中含有某一字符串的图片存到指定文件夹下
|
目的：
某一个文件下存放的是各种各样所有的图片，希望把这些图片进行分类。比如：将图片名字中含有绝缘子自爆的图片放在绝缘子自爆的文件夹下。将图片名字中含有树枝的图片放在树枝的文件夹下。 一、将含有某一个特定字符串的图片的完整名字存放在一个.txt文本下 """ 打开图片文件夹，获取文件夹中图片的名字，如果图片名字中包含某个字符串，把这个图片的完整名字存在一个.txt文本中 参考：https://blog.csdn.net/zhangxiangweide/article/details/78857997 修改两处绝缘子自爆，注意自己在相应位置 建立文件夹 """ # -*- coding:utf-8 -*- import os #os：操作系统相关的信息模块 #存放原始图片地址 data_base_dir = "/home/dlut/网络/make_database/全部巡视缺陷图片/"#含所有图片的文件 file_list = [] #建立列表，用于保存图片信息 #读取图片文件，并将图片地址、图片名和标签写到txt文件中 write_file_name = '/home/dlut/网络/make_database/缺陷图片分类/绝缘子自爆.txt' #自己手动创建一个.txt文本 write_file = open(write_file_name, "w") #以只写方式打开write_file_name文件 for file in os.listdir(data_base_dir): #file为current_dir当前目录下图片名 if file.find("绝缘子自爆")>0: #需要找带有什么名字的字符串 """ find函数：如果包含子字符串返回开始的索引值，否则返回-1。所以我用了一个判断语句，判断是否大于0，大于0，说明包含这个字符串，保存名字。 如果不写大于0，不论是否包含绝缘子这几个字，都会把图片保存下来。 if语句 0为false 其他数字均为true """ write_name = file #图片路径 + 图片名 + 标签 file_list.append(write_name) #将write_name添加到file_list列表最后 number_of_lines = len(file_list) #列表中元素个数 for current_line in range(number_of_lines): write_file.write(file_list[current_line] + '\n') write_file.close()结果：（截图未截全） 二、将txt文件中的名字的图片存在另一个自己新建的文件夹下，同时在原文件夹下删除这些图片 ##########################某一个txt文本中的数字存的是图片的名字，要把这些名字的图片保存到另一个文件夹中,同时删除了原文件下的这些图片（可以自己选择不删除os.remove）######################## #修改两处，注意自己建立文件 from PIL import Image import os f3 = open("/home/dlut/网络/make_database/缺陷图片分类/绝缘子自爆.txt",'r') #test文件所在路径 for line2 in f3.readlines(): line3=line2[:-5] #读取每行去掉后四位的数 im = Image.open('/home/dlut/网络/make_database/全部巡视缺陷图片/{}.JPG'.format(line3))#打开改路径下的line3记录的的文件名 im.save('/home/dlut/网络/make_database/缺陷图片分类/绝缘子自爆/{}.JPG'.format(line3)) #把文件夹中指定的文件名称的图片另存到该路径下 #注意！！！！！！！下面这句话是删除原文件下含有这些名字的图片，也可以选择不删除，那就删掉下面那句话 os.remove('/home/dlut/网络/make_database/全部巡视缺陷图片/{}.JPG'.format(line3)) #删除原来文件夹下名字包括绝缘子自爆的图片 f3.close()结果：（截图未截全） 参考： https://blog.csdn.net/zhangxiangweide/article/details/78857997 https://blog.csdn.net/qq_39418067/article/details/89048165 https://blog.csdn.net/yongwan5637/article/details/80622009 https://blog.csdn.net/weixin_43384257/article/details/98374743 |
【本文地址】
今日新闻 |
推荐新闻 |