for文を回してファイルに画像を保存したい

Question

for文を回して、それぞれクラスタリングされたファイルに画像を保存したい

コーディネートを分類するアルゴリズムを作っています。
コーディネートをクラスタリングするところまではうまくできました。特徴をもとに５つのグループに分けています。
そこで次は、それぞれのコーディネート画像をクラスタリングされたグループのファイルに保存したいです。

import os
import shutil
import numpy as np
from PIL import Image
from skimage import data
from sklearn.cluster import KMeans
import glob
# 1. 3次元配列の画像データを2次元配列のデータに変換
search_dir = './img/img_convert'
feature = np.array([
data.imread(path)
for path in glob.glob('./img/img_convert/*.jpg')
])
feature = feature.reshape(len(feature), -1).astype(np.float64)
# 2. 学習(5種類のグループにクラスタリングする)
model = KMeans(n_clusters=5).fit(feature)
# 3. 学習結果のラベル
labels = model.labels_
# 4. 学習結果(クラスタリング結果の表示 + ラベルごとにフォルダ分け)
for label, path in zip(labels, os.listdir('./img/img_convert')):
    if label==0:
        shutil.move(path,'./img/img_group/0')
    elif label==1:
        shutil.move(path,'./img/img_group/1')
    elif label==2:
        shutil.move(path,'./img/img_group/2')
    elif label==3:
        shutil.move(path,'./img/img_group/3')
    elif label==4:
        shutil.move(path,'./img/img_group/4')
    print(label, path)

これで実行すると、
FileNotFoundError: [Errno 2] No such file or directory: '20180906060309102_276.jpg'
というエラーが出ます。
どなたかやり方をご教授ください。

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

python3.6.6

Comments

[退会済みユーザー]

python switchで調べてみては

Answer 1

FileNotFoundError: [Errno 2] No such file or directory: '20180906060309102_276.jpg'

ファイルが存在しないからかと。
コンパイル通してませんが。

import os.path
for label, path in zip(labels, os.listdir('./img/img_convert')):
    shutil.move(os.path.join(search_dir,path), './img/img_group/' + str(label))

os.listdirの並び順は保証されていないので、ソートしてくださいな。

---

Error: Destination path './img/img_group/3/.ipynb_checkpoints' already existsError: Destination path './img/img_group/3/.ipynb_checkpoints' already exists

抽出条件が違うからです。

処理	抽出条件
glob.glob	指定したディレクトリ内のjpgファイルのみ
os.listdir	指定したディレクトリ内のファイル全て

glob.globの結果をリストに退避しておけばよいかと。
並び順を保証するソートにも対応したバージョンです。

# -*- coding: utf-8 -*-
import os
import shutil
import numpy as np
from PIL import Image
from skimage import data
from sklearn.cluster import KMeans
import glob
from pathlib import Path
# 1. 3次元配列の画像データを2次元配列のデータに変換
search_dir = './img/img_convert'
# ソート
search_data = sorted(glob.glob(search_dir + '/*.jpg'))
feature = np.array([data.imread(path) for path in search_data])
feature = feature.reshape(len(feature), -1).astype(np.float64)
# 2. 学習(5種類のグループにクラスタリングする)
model = KMeans(n_clusters=5).fit(feature)
# 3. 学習結果のラベル
labels = model.labels_

#
move_dir = Path('./img/img_group/')
# ディレクトリ作成
move_dir.mkdir(parents=True, exist_ok=True)
for label, path in zip(labels, search_data):
    src_file = Path(search_dir, path)
    shutil.move(str(src_file),  str(move_dir / str(label)))

---

glob.globはPathlib#globで代替できるので、shutilやglobを使わずにPathlibモジュールに統一することをお勧めいたします。

Comments

[matt17]

ご回答ありがとうございます。
ファイルごとに保存することができました！
しかし、
Error: Destination path './img/img_group/3/.ipynb_checkpoints' already existsError: Destination path './img/img_group/3/.ipynb_checkpoints' already exists
というエラーが出てしまいます。
分けられているので機能的には問題ないのですが、これはどういうことなのでしょうか。

[matt17]

ご回答ありがとうございます。
うまく行きました！本当にありがとうございました！

[matt17]

ご回答ありがとうございます。
うまく行きました！本当にありがとうございました！

[umyu]

@matt17さんへ
解決して良かったです。質問文のプログラムを今後も使う予定ならばPathlib https://docs.python.jp/3/library/pathlib.html について調べてみるのをお勧めします。