Python从位于url的图像计算phash

2024-04-25 06:45:33 发布

您现在位置:Python中文网/ 问答频道 /正文

我想从大约10.000.000张图片中计算PHAH,我只知道它们所在的url。

我知道如何下载一张图片,然后计算相位,但我总是要先确保图片的安全

是否可以下载图片并计算PHAH而不保存它,或者甚至可以不下载图片并仅使用url计算PHAH?

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

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")

Tags: csvpathimageurldataoscounter图片
1条回答
网友
1楼 · 发布于 2024-04-25 06:45:33

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

下面是代码中的外观:

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

相关问题 更多 >