adien / phash.py

#!/usr/bin/env -S uv run --script # /// script # requires-python = ">=3.12" # dependencies = [ # "imagehash" # ] # /// from PIL import Image import imagehash import numpy def from64bit_hash(value: int): """ Converts a 64-bit unsigned long integer into an ImageHash object. The integer is transformed into a 8x8 binary numpy array, which is the standard representation for many perceptual hashes. :param value: The 64-bit integer hash value. Must be between 0 and 2**64 - 1. :return: An ImageHash object representing the integer. """ if not (0 <= value < 2**64): raise ValueError("Value must be a 64-bit unsigned integer (between 0 and 2**64 - 1).") # 1. Convert the integer to its 64-bit binary string representation. # numpy.binary_repr is perfect for this, as it handles padding. # Example: 1 -> '000...001' binary_string = numpy.binary_repr(value, width=64) # 2. Convert the binary string into a numpy array of booleans. # '1' becomes True, '0' becomes False. binary_array = numpy.array([char == '1' for char in binary_string], dtype=bool) # 3. Reshape the 64-element flat array into a 8x8 matrix. matrix = binary_array.reshape((8, 8), order='F') # 4. Create and return the ImageHash instance. return imagehash.ImageHash(matrix) if __name__ == '__main__': result = imagehash.phash(Image.open('pepper.png')) result2 = from64bit_hash(15500626565295817037) print(result - result2)