Auf Dokumente Logo erkennen

import cv2

import numpy as np

import os

import shutil

def find_logo(target_image_path, logo_image_gray, search_area=(800, 400), threshold=0.8):

    """Search for the logo within the specified area of the target image, focusing on the top right corner and handling grayscale images."""

    # Load the target image and convert to grayscale

    target_image = cv2.imread(target_image_path)

    target_image_gray = cv2.cvtColor(target_image, cv2.COLOR_BGR2GRAY)

    height, width = target_image_gray.shape

    # Calculate the starting point for the crop to focus on the top right corner

    start_x = width - search_area[0]  # Width of the image minus the search area width

    start_y = 0  # Start at the top of the image

    # Crop the search area from the top right corner of the target image

    search_area_cropped = target_image_gray[start_y:start_y + search_area[1], start_x:start_x + search_area[0]]

    # Template matching

    result = cv2.matchTemplate(search_area_cropped, logo_image_gray, cv2.TM_CCOEFF_NORMED)

    _, max_val, _, _ = cv2.minMaxLoc(result)

    return max_val >= threshold

def process_images(folder_path, logo_path, similar_folder="similar", threshold=0.8):

    """Process all images in the folder to find the logo within the top right area, considering grayscale images."""

    # Convert the reference (logo) image to grayscale once

    logo_image = cv2.imread(logo_path)

    logo_image_gray = cv2.cvtColor(logo_image, cv2.COLOR_BGR2GRAY)

    # Ensure the 'similar' folder exists

    similar_folder_path = os.path.join(folder_path, similar_folder)

    if not os.path.exists(similar_folder_path):

        os.makedirs(similar_folder_path)

    # Iterate through all images in the folder

    for filename in os.listdir(folder_path):

        if filename.endswith(('.png', '.jpg', '.jpeg')):

            current_path = os.path.join(folder_path, filename)

            try:

                if find_logo(current_path, logo_image_gray, threshold=threshold):

                    # If the logo is found, move the image to the 'similar' folder

                    shutil.move(current_path, os.path.join(similar_folder_path, filename))

                    print(f"Moved {filename} to {similar_folder_path}.")

            except Exception as e:

                print(f"Error processing {filename}: {e}")

# Example usage

folder_path = 'path/to/your/pictures'  # Update this to your pictures folder path

logo_path = 'logo.jpg'  # Ensure this is the path to your logo

process_images(folder_path, logo_path)