Script para identificar diferenças entre imagens ou buscar imagens iguais em uma árvore de diretórios

image_comparison.py

import os
import sys
import cv2


def check_differences():
    im1 = cv2.imread(sys.argv[1])
    im2 = cv2.imread(sys.argv[2])

    diff = cv2.subtract(im1, im2)

    print(f"As imagens {sys.argv[1]} e {sys.argv[2]} são ", end="")
    if diff.any():
        print("diferentes.")
        cv2.imshow("Diferencas encontradas", diff)
        cv2.waitKey(5000)
    else:
        print("iguais.")


def are_equal():
    print(cv2.subtract(cv2.imread(sys.argv[1]), cv2.imread(sys.argv[2])).any())


def find_equal():
    im1 = cv2.imread(sys.argv[1])
    for dirpath, _, filenames in os.walk(sys.argv[2]):
        for filename in filenames:
            path = os.path.join(dirpath, filename)
            if filename.endswith(".png"):
                try:
                    im2 = cv2.imread(path)
                    diff1 = cv2.subtract(im1, im2)
                    diff2 = cv2.subtract(im2, im1)
                    if not diff1.any() and not diff2.any():
                        print(path)
                except cv2.error:
                    pass


if __name__ == '__main__':
    find_equal()