removed searchOne
parent
8b249f7f25
commit
1e253f2b5b
@ -1,46 +0,0 @@
|
||||
import os
|
||||
import imagehash
|
||||
from PIL import Image
|
||||
import shutil
|
||||
|
||||
# Paths for your folders and Waldo image
|
||||
input_folder = "images" # Folder with images to check for Waldo
|
||||
output_folder = "found_waldo_images" # Folder to save images containing Waldo
|
||||
waldo_image_path = "symposium/waldo_transparent.png" # Path to your Waldo image
|
||||
|
||||
# Create output folder if it doesn't exist
|
||||
os.makedirs(output_folder, exist_ok=True)
|
||||
|
||||
# Load the Waldo image and compute its hash
|
||||
waldo_image = Image.open(waldo_image_path)
|
||||
waldo_hash = imagehash.phash(waldo_image)
|
||||
|
||||
# Function to compare the hash of an image with Waldo's hash
|
||||
def detect_waldo_in_image(image_path):
|
||||
try:
|
||||
# Open image
|
||||
image = Image.open(image_path)
|
||||
# Compute the hash of the current image
|
||||
image_hash = imagehash.phash(image)
|
||||
|
||||
# Compute the difference between the image hash and Waldo's hash
|
||||
hash_diff = waldo_hash - image_hash
|
||||
|
||||
# If the difference is small (threshold), we consider it a match
|
||||
return hash_diff < 13 # You can adjust the threshold (10) based on your needs
|
||||
except Exception as e:
|
||||
print(f"Error processing {image_path}: {e}")
|
||||
return False
|
||||
|
||||
# Process each image in the input folder
|
||||
for filename in os.listdir(input_folder):
|
||||
if filename.lower().endswith(('.png', '.jpg', '.jpeg', '.bmp', '.tiff')): # Process common image formats
|
||||
image_path = os.path.join(input_folder, filename)
|
||||
|
||||
# Check if the image matches the Waldo pattern
|
||||
if detect_waldo_in_image(image_path):
|
||||
print(f"Found Waldo in {filename}, saving to {output_folder}")
|
||||
# Copy the image to the output folder
|
||||
shutil.copy(image_path, os.path.join(output_folder, filename))
|
||||
|
||||
print("Processing complete. Images with Waldo have been saved.")
|
Loading…
Reference in New Issue