Python 从位于 url 的图像计算 phash

Question

我想从大约 10.000.000 张图片计算 phash，从中我只有 url 它们所在的位置。

我知道如何下载图片然后计算phash，但我总是必须先保护图片。

是否可以下载图片并在不保存的情况下计算 phash，或者甚至可以根本不下载图片而只使用 url 计算 phash？

这是我下载前十张图片并计算 phash 的代码：

folder, pic_savefolder = 'data', 'data/pictures'
file = 'external-asset-url-clean.csv'
path = os.path.join(folder,file)
df = pd.read_csv(path, header=None, names=["URL"])
counter = 0
hashes = set()
headers = {"User-Agent":"Mozilla/5.0 (Macintosh; Intel Mac OS X 10_15_4) AppleWebKit/537.36 (KHTML, like Gecko) Chrome/83.0.4103.97 Safari/537.36"}

for image_url in df['URL']:
    filename = image_url.split('/')[-1]
    try:
        r = requests.get(image_url, allow_redirects=False,verify=False, headers=headers)
        pathlong = os.path.join(pic_savefolder, filename)
        with open(pathlong,"wb") as f:
            f.write(r.content)
            hash = imagehash.phash(Image.open(pathlong))
            hashes.add((hash))
        counter += 1
        if counter > 10:
            break
except Exception as e:
    print(e)
    print("\n")

Answer 1

如果您使用 .raw 属性而不是 .content，则可以直接传递内容而不是写入文件。

代码如下所示：

image_data = Image.open(requests.get(image_url, stream=True).raw)
hash = imagehash.phash(image_data)

Python 从位于 url 的图像计算 phash

Python calculate phash from an image located at a url

python

phash

python-requests

pytorch

imagehash