Python 画像分類条件の average_hash 値を変更したい

Question

pythonの画像分類に興味を持ち過去の質問を調べましたが、
見つからなかったため、教えて欲しいです。

１つのフォルダ内の似た画像のファイル名をプリントしたいです。
imagehash(average_hash)を使った分類方法です。

下記コードで同一画像がプリント出来るようですが、
希望としては、average_hash 値<10 をプリントしたいです。
お手数おかけしますが、ご指導願います。

前提条件
５つの似た test-1〜5.jpg という名の画像が
"a"という名のフォルダに入っています。
test-1.jpg と test-4.jpg は全く同じ画像です。

該当のソースコード

import os
from PIL import Image
import imagehash

userpath = './a'  # 検索フォルダ名'a'

image_files = []
f = [os.path.join(userpath, path) for path in os.listdir(userpath)]
for i in f:
    if i.endswith('.jpg') or i.endswith('.png'):
        image_files.append(i)

imgs = {}
for img in sorted(image_files):
    hash = imagehash.average_hash(Image.open(img))
    if hash in imgs:
        print('Similar image :', img, imgs[hash])
    else:
        imgs[hash] = img

#print 結果：　Similar image : ./a\\test-4.jpg ./a\test-1.jpg

試したこと

下記コードの末尾に<10を入力しましたがエラーとなりました。
hash = imagehash.average_hash(Image.open(img))<10

補足情報（FW/ツールのバージョンなど）

python 3.7.7
ImageHash 4.3.1

Comments

[meg_]

該当のソースコード の引用元はどこでしょうか？質問に追記しましょう。

[TUTINO]

ご確認いただきありがとうございます。
このコードは友達から送ってもらいましたので、確認いたします。
少々お待ち願います。

[meg_]

コードにも著作権がありますので、他者のコードを引用される際には引用元を追記された方が良いかと思います。

[TUTINO]

著作権についてもアドバイスいただき、ありがとうございます。今後は十分に注意いたします。

Answer 1

下記の様なコードでどうでしょうか？

import os
from PIL import Image
import imagehash

userpath = './a'  # 検索フォルダ名'a'

image_files = []
f = [os.path.join(userpath, path) for path in os.listdir(userpath)]
for i in f:
    if i.endswith('.jpg') or i.endswith('.png'):
        image_files.append(i)

image_files = sorted(image_files)
for i, img in enumerate(image_files):         
    base_hash = imagehash.average_hash(Image.open(img))
    for j in range(i + 1, len(image_files)):
        other_img = image_files[j]
        if img == other_img:
            continue
        other_hash = imagehash.average_hash(Image.open(other_img))
        
        if abs(base_hash - other_hash) < 10:
            print(f"imagehash値の差 10より小さい：画像1 {img}, 画像2 {other_img}")

Comments

[TUTINO]

この短時間で解決いただき、本当にありがとうございました。
WEBで１日調べても分からず諦めかけてました。
重ねて感謝いたします。