【python】删除重复图片

源码 3年前 0 209

今天闲来无事整理素材，发现有许多重复的图片，由于数量太多无法手动翻阅删除，想想写个代码，分析了一下重复图片有些是同名的，有些内容重复不同名，返回文件名清理放弃，图片大小也放弃放弃，后来选用计算MD5的方式清除，然后先是使用os.listdir()函数遍历文件夹下的图片，在测试过程中，如果文件夹下还包含文件夹就会引发异常，后来决定用os.walk()函数来遍历；

源码注释写的比较清晰，就不多阐述，可根据自己需要封装函数，制作成死循环，添加退出条件，复用；
部分源码展示
[Python] 纯文本查看 复制代码try:
file = os.walk(path) # 递归遍历目录；
except FileNotFoundError: # 捕获路径不存在异常；
print(\’抱歉，没有这个路径！\’)
else:
temp = set() # 创建临时集合；
del_count = 0 # 删除图片计数；
pass_count = 0 # 非图片计数；
file_count = 0 # 总文件计数；
time1 = time()
for path_name, dir_name, file_name in file: # 遍历walk返回3个元素；
for n in file_name: # 获得每个文件名字；
full_path = os.path.join(path_name, n) # 拼接路径和文件名，获得文件完整路径；
file_count += 1 # 文件计数+1；
print(full_path)
try:
with Image.open(full_path) as t: # 打开图片；
array = np.array(t) # 转为数组；
except (UnidentifiedImageError,DecompressionBombError): # 捕获不是图片，像素炸弹异常；
pass_count += 1 # 非图片计数+1；
pass
else:
md5 = hashlib.md5() # 创建MD5对象；
md5.update(array) # 获取当前图片MD5；
if md5.hexdigest() not in temp: # 如果哈希值没有在集合中；
temp.add(md5.hexdigest()) # 就把哈希值添加到集合中；
else:
os.remove(full_path) # 如果在集合中就删除当前图片；
print(full_path+\’——————已删除\’)
del_count += 1 # 删除计数+1