from PIL import Image, ImageChops
def compare_images(path1, path2, diff_save_location):
"""
比较两个图像并保存差异图像。
参数:
- path1: 第一个图像的路径。
- path2: 第二个图像的路径。
- diff_save_location: 差异图像保存的路径。
"""
image1 = Image.open(path1)
image2 = Image.open(path2)
if image1.mode != image2.mode:
image2 = image2.convert(image1.mode)
if image1.size != image2.size:
print("Images have different sizes.")
return
diff = ImageChops.difference(image1, image2)
if diff.getbbox():
diff = ImageChops.add(diff, diff, 2.0, -100)
diff.save(diff_save_location)
print(f"Difference image saved to {diff_save_location}")
compare_images("./val/images/9-R.bmp", "./val/NG/9-R.bmp", "diff.png")
判断两幅图像是否一致,并保存两幅图像的差异图
最新推荐文章于 2024-07-26 17:36:26 发布