Spaces:
Sleeping
Sleeping
File size: 1,903 Bytes
22e1b62 |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 |
from google_img_source_search import ReverseImageSearcher
import requests
from io import BytesIO
from PIL import Image
import imagehash
from google_img_source_search import ReverseImageSearcher
def get_image_from_url(url):
response = requests.get(url)
return Image.open(BytesIO(response.content))
def standardize_image(image):
# Convert to RGB if needed
if image.mode in ('RGBA', 'LA'):
background = Image.new('RGB', image.size, (255, 255, 255))
background.paste(image, mask=image.split()[-1])
image = background
elif image.mode != 'RGB':
image = image.convert('RGB')
# Resize to standard size (e.g. 256x256)
standard_size = (256, 256)
image = image.resize(standard_size)
return image
def compare_images(image1, image2):
# Standardize both images first
img1_std = standardize_image(image1)
img2_std = standardize_image(image2)
hash1 = imagehash.average_hash(img1_std)
hash2 = imagehash.average_hash(img2_std)
return hash1 - hash2 # Returns the Hamming distance between the hashes
if __name__ == '__main__':
image_url = 'https://i.pinimg.com/originals/c4/50/35/c450352ac6ea8645ead206721673e8fb.png'
# Get the image from URL
url_image = get_image_from_url(image_url)
# Search image
rev_img_searcher = ReverseImageSearcher()
res = rev_img_searcher.search(image_url)
for search_item in res:
print(f'Title: {search_item.page_title}')
# print(f'Site: {search_item.page_url}')
print(f'Img: {search_item.image_url}\n')
# Compare each search result image with the input image
result_image = get_image_from_url(search_item.image_url)
result_difference = compare_images(result_image, url_image)
print(f"Difference with search result: {result_difference}")
if result_difference == 0:
break |