質問をすることでしか得られない、回答やアドバイスがある。

15分調べてもわからないことは、質問しよう!

ただいまの
回答率

89.99%

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

解決済

回答 1

投稿 編集

  • 評価
  • クリップ 0
  • VIEW 665

matt17

score 14

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

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

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

  • 気になる質問をクリップする

    クリップした質問は、後からいつでもマイページで確認できます。

    またクリップした質問に回答があった際、通知やメールを受け取ることができます。

    クリップを取り消します

  • 良い質問の評価を上げる

    以下のような質問は評価を上げましょう

    • 質問内容が明確
    • 自分も答えを知りたい
    • 質問者以外のユーザにも役立つ

    評価が高い質問は、TOPページの「注目」タブのフィードに表示されやすくなります。

    質問の評価を上げたことを取り消します

  • 評価を下げられる数の上限に達しました

    評価を下げることができません

    • 1日5回まで評価を下げられます
    • 1日に1ユーザに対して2回まで評価を下げられます

    質問の評価を下げる

    teratailでは下記のような質問を「具体的に困っていることがない質問」、「サイトポリシーに違反する質問」と定義し、推奨していません。

    • プログラミングに関係のない質問
    • やってほしいことだけを記載した丸投げの質問
    • 問題・課題が含まれていない質問
    • 意図的に内容が抹消された質問
    • 広告と受け取られるような投稿

    評価が下がると、TOPページの「アクティブ」「注目」タブのフィードに表示されにくくなります。

    質問の評価を下げたことを取り消します

    この機能は開放されていません

    評価を下げる条件を満たしてません

    評価を下げる理由を選択してください

    詳細な説明はこちら

    上記に当てはまらず、質問内容が明確になっていない質問には「情報の追加・修正依頼」機能からコメントをしてください。

    質問の評価を下げる機能の利用条件

    この機能を利用するためには、以下の事項を行う必要があります。

質問への追記・修正、ベストアンサー選択の依頼

  • 退会済みユーザー

    退会済みユーザー

    2018/09/06 23:36

    python switchで調べてみては

    キャンセル

回答 1

checkベストアンサー

0

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.globPathlib#globで代替できるので、shutilglobを使わずにPathlibモジュールに統一することをお勧めいたします。

投稿

編集

  • 回答の評価を上げる

    以下のような回答は評価を上げましょう

    • 正しい回答
    • わかりやすい回答
    • ためになる回答

    評価が高い回答ほどページの上位に表示されます。

  • 回答の評価を下げる

    下記のような回答は推奨されていません。

    • 間違っている回答
    • 質問の回答になっていない投稿
    • スパムや攻撃的な表現を用いた投稿

    評価を下げる際はその理由を明確に伝え、適切な回答に修正してもらいましょう。

  • 2018/09/07 13:01

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

    キャンセル

  • 2018/09/07 13:10

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

    キャンセル

  • 2018/09/07 13:10

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

    キャンセル

  • 2018/09/07 13:13

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

    キャンセル

15分調べてもわからないことは、teratailで質問しよう!

  • ただいまの回答率 89.99%
  • 質問をまとめることで、思考を整理して素早く解決
  • テンプレート機能で、簡単に質問をまとめられる