近期在整理相册的时候,发现相册中有许多重复图片,人工一张张筛查删除太枯燥,便写下这个脚本,用于删除文件夹下重复的图片。
第一部分:判断两张图片是否相同
要查找重复的图片,必然绕不开判断两张图片是否相同。判断两张图片简单呀!图片可以看成数组,比较两个数组是否相等不就行了。但是这样做太过简单粗暴,因为两个数组的每个元素都要一一比较,效率很低。为了尽量避免两个庞大的数组比较:
- 先进行两张图片的大小(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
来源:https://www.cnblogs.com/aiyou-3344520/p/14760053.html
图文来源于网络,如有侵权请联系删除。