以下のサイトを参考に、自身で用意した画像をクラスタリングするプログラムを作成しています。
scikit-learnで国旗画像をクラスタリングして似ているものを探す
このサイトのままだとfrom skimage import dataの部分で既に用意してある国旗の画像を読み込むだけみたいなので以下のようにしました。
自身で用意した画像が入ってるフォルダはphotoというフォルダ名にしてあります。
python
1import cv2 2import os 3import shutil 4import numpy as np 5from PIL import Image 6from skimage import data 7from sklearn.cluster import KMeans 8 9# 1. 放電画像のサイズをそろえて保存する 10# ./photo 以下に放電画像 11# ./photo 以下に100*100のサイズに変換したjpgを保存 12for path in os.listdir('./photo'): 13 img = Image.open(f'./photo/{path}') 14 img = img.convert('RGB') 15 img_resize = img.resize((100, 100)) 16 img_resize.save(f'./copy/{path}.jpg') 17 18# 2. 3次元配列の画像データを2次元配列のデータに変換 19for num in range(10): 20 feature = cv2.imread(f'./copy/{path}') 21 print(feature.shape) 22 feature = feature.reshape(len(feature), -1).astype(np.float64) 23 print(feature.shape) 24 25# 3. 学習(2種類のグループにクラスタリングする) 26model = KMeans(n_clusters=2).fit(feature) 27 28# 4. 学習結果のラベル 29labels = model.labels_ 30 31# 5. 学習結果(クラスタリング結果の表示 + ラベルごとにフォルダ分け) 32# ./flag_group 以下に画像を分けて保存する 33for label, path in zip(labels, os.listdir('./copy')): 34 os.makedirs(f"./flag_group/{label}", exist_ok=True) 35 shutil.copyfile(f"./photo/{path.replace('.jpg', '')}", f"./flag_group/{label}/{path.replace('.jpg', '')}") 36 print(label, path)
すると35行目で以下のようなエラーが出ました
例外が発生しました: FileNotFoundError
[Errno 2] No such file or directory: './flag_origin/sioari (1)'
sioari(1)というのは自分が用意したフォルダに入ってる画像の内の一枚です。
ここからどうすれば解決でき、フォルダ内の画像をサイトのようにクラスタリングできるか教えていただきたく思います。
よろしくお願いいたします。
回答1件
あなたの回答
tips
プレビュー