前の質問の後色々調べてみたところ、やはりおかしいところはないように思えたので質問させていただきます
画像を用意してnp.saveで学習用のファイルを作ろうとしています。
該当のソースコード
python
1from PIL import Image 2import os, glob 3import numpy as np 4import random, math 5 6#学習用画像が保存されているディレクトリ 7train_dir = "E:\python\monster_py\monster_py" 8 9#テスト用画像が保存されているディレクトリ 10test_dir = "E:\python\dis_download" 11 12# モンスターの種類 13categories = ["モンスターエナジー","モンスターエナジー パイプラインパンチ" 14,"モンスター キューバリブレ","モンスターエナジー アブソリュートリーゼロ" 15,"モンスターエナジー カオス","モンスターエナジー ウルトラパラダイス","モンスターエナジー ウルトラ"] 16# 画像データ用配列 17X = [] 18# ラベルデータ用配列 19Y = [] 20 21#画像データごとにadd_sample()を呼び出し、X,Yの配列を返す関数 22def make_sample(files): 23 global X, Y 24 X = [] 25 Y = [] 26 for cat, fname in files: 27 add_sample(cat, fname) 28 return np.array(X), np.array(Y) 29 30def add_sample(cat, fname): 31 img = Image.open(fname) 32 img = img.convert("RGB") 33 img = img.resize((150, 150)) 34 data = np.asarray(img) 35 X.append(data) 36 Y.append(cat) 37 38#学習用画像 39trainfiles = [] 40#テスト用画像 41testfiles = [] 42 43for idx, cat in enumerate(categories): 44 image_dir = train_dir + "/" + cat 45 files = glob.glob(image_dir + "/*.jpg") 46 for f in files: 47 trainfiles.append((idx, f)) 48 49for idx, cat in enumerate(categories): 50 image_dir = test_dir + "/" + cat 51 files = glob.glob(image_dir + "/*.jpg") 52 for f in files: 53 testfiles.append((idx, f)) 54 55X_train, y_train = make_sample(trainfiles) 56X_test, y_test = make_sample(testfiles) 57xy = (X_train, X_test, y_train, y_test) 58print(xy) 59#データを保存する 60np.save("E:\python\monster_data.npy", xy)
print(xy)の結果は
python
1(array([], dtype=float64), array([], dtype=float64), array([], dtype=float64), array([], dtype=float64))
です。arrayに何も入っていない理由がわかる方いましたら教えて下さい
###参考にしたサイト
https://qiita.com/kojikoji_ksnv/items/12dfe9075cb1dd3afe59
trainfiles、testfilesを初期化した後、処理前後のタイミングで変数の値を確認してみてください。問題の箇所が判るかと思います。
すみません、初心者なもので変数値の確認方法がわかりません。どのようにしたら良いのでしょうか
print(trainfiles)、print(testfiles)で良いのではないでしょうか?
※デバッグ方法は「python、デバッグ」などで調べてみてください。調べてトライ&エラーを繰り返すことでプログラムが書けるようになります。
vscのデバッグ機能を使い調べましたがエラーは出ませんでした
また、printをしてみたところどの位置でもarrayは値なしでした
「デバッグ」についてはそういう意味(エラーが出る)で言ったのではありませんでした。プログラムが意図通り動作しないときには、デバッグすることが大事、という意味でした。
for文の後でも[]のままということはforの中の処理が間違っているのでしょう。
filesの値はjpgファイルの一覧になっていますか?
files=~~はimage_dir(学習用フォルダパス)になっています。拡張子も確認しましたがJPGでした
filesが「jpgファイルへのパス」のリストになっているのであれば、trainfilesが空のリストのままにはならないはずです。
> また、printをしてみたところどの位置でもarrayは値なしでした
というのは本当ですか?
train,testディレクトリにそれぞれcategory7種のフォルダを作り、そこに画像を入れたら最後まで問題なく実行できました、ありがとうございました。
回答1件
あなたの回答
tips
プレビュー