Python脚本:删除文件夹下的重复图片,实现图片去重

Python脚本:删除文件夹下的重复图片,实现图片去重

 

近期在整理相册的时候,发现相册中有许多重复图片,人工一张张筛查删除太枯燥,便写下这个脚本,用于删除文件夹下重复的图片。

 

第一部分:判断两张图片是否相同

要查找重复的图片,必然绕不开判断两张图片是否相同。判断两张图片简单呀!图片可以看成数组,比较两个数组是否相等不就行了。但是这样做太过简单粗暴,因为两个数组的每个元素都要一一比较,效率很低。为了尽量避免两个庞大的数组比较:

  • 先进行两张图片的大小(byte)比较,若大小不相同,则两张图片不相同;
  • 在两张图片的大小相同的前提下,进行两张图片的尺寸(长和宽)比较,若尺寸不相同,则两张不相同;
  • 在两张图片的尺寸相同的前提下,进行两张图片的内容(即数组元素)比较,若内容不相同,则图片不相同;

这样,当图片大小或图片尺寸不相同的时候,便认为两张图片不同,可以省去比较数组元素的部分,效率up~

 1 import shutil
 2 import numpy as np
 3 from PIL import Image
 4 import os
 5 
 6 
 7 def 比较图片大小(dir_image1, dir_image2):
 8     with open(dir_image1, "rb") as f1:
 9         size1 = len(f1.read())
10     with open(dir_image2, "rb") as f2:
11         size2 = len(f2.read())
12     if(size1 == size2):
13         result = "大小相同"
14     else:
15         result = "大小不同"
16     return result
17 
18 
19 def 比较图片尺寸(dir_image1, dir_image2):
20     image1 = Image.open(dir_image1)
21     image2 = Image.open(dir_image2)
22     if(image1.size == image2.size):
23         result = "尺寸相同"
24     else:
25         result = "尺寸不同"
26     return result
27 
28 
29 def 比较图片内容(dir_image1, dir_image2):
30     image1 = np.array(Image.open(dir_image1))
31     image2 = np.array(Image.open(dir_image2))
32     if(np.array_equal(image1, image2)):
33         result = "内容相同"
34     else:
35         result = "内容不同"
36     return result
37 
38 
39 def 比较两张图片是否相同(dir_image1, dir_image2):
40     # 比较两张图片是否相同
41     # 第一步:比较大小是否相同
42     # 第二步:比较长和宽是否相同
43     # 第三步:比较每个像素是否相同
44     # 如果前一步不相同,则两张图片必不相同
45     result = "两张图不同"
46     大小 = 比较图片大小(dir_image1, dir_image2)
47     if(大小 == "大小相同"):
48         尺寸 = 比较图片尺寸(dir_image1, dir_image2)
49         if(尺寸 == "尺寸相同"):
50             内容 = 比较图片内容(dir_image1, dir_image2)
51             if(内容 == "内容相同"):
52                 result = "两张图相同"
53     return result
hmoban主题是根据ripro二开的主题,极致后台体验,无插件,集成会员系统
自学咖网 » Python脚本:删除文件夹下的重复图片,实现图片去重