初心者で機械学習でのエラーの解決ができなかったので教えてください！

Question

前提

Google colaboratryを使用して機械学習を勉強してました。機械学習は始めたてなこともあり、あるサイトを参考にしながらプログラムを書いていました。まだまだ分からないことも多いためエラーを解決することができなかったのでどなたか解決策を教えて頂けると幸いです…

実現したいこと

画像を認識しフォルダー分けをする

発生している問題・エラーメッセージ

NameError: name 'pred' is not defined

該当のソースコード

from keras.models import load_model
import pickle
import cv2
import keras

model = keras.models.load_model('/content/drive/MyDrive/Colab Notebooks/機械学習/model.h5', compile=False)
classes = pickle.load(open('/content/drive/MyDrive/Colab Notebooks/機械学習/classes.sav', 'rb'))

img = cv2.imread('/content/drive/MyDrive/Colab Notebooks/機械学習/sample/sample.jpg')
img = cv2.resize(img,dsize=(224,224))
img = img.astype('float32')
img /= 255.0
img = img[None, ...]

img = cv2.imread('/content/drive/MyDrive/Colab Notebooks/機械学習/sample/sample.jpg')
cv2.imwrite('/content/drive/MyDrive/Colab Notebooks/機械学習/output/' + str(classes[pred])+ '/sample.jpg',img)

試したこと

エラー内容を調べ理解まではしましたが解決はできませんでした。

補足情報

こちらのサイトを参考にしました。
https://sasuwo.org/image-classification/#toc3

Comments

[hoshi-takanori]

たぶん元の記事のコードに
result = model.predict(img)
pred = result.argmax()
の 2 行が抜けてる気が…。

[Yukkiy]

そうなんですね。定義されてないよって言われてたのでそんな気はしてました。ちなみにどこでその変数指定すればいいですか…？

Answer 1

参考サイト見てみましたが、あまり初心者が見るには親切なサイトではないですね。。。
下記で動作すると思います

from keras.models import load_model
import pickle
import cv2
import keras

model = keras.models.load_model('/content/drive/MyDrive/Colab Notebooks/機械学習/model.h5', compile=False)
classes = pickle.load(open('/content/drive/MyDrive/Colab Notebooks/機械学習/classes.sav', 'rb'))

img = cv2.imread('/content/drive/MyDrive/Colab Notebooks/機械学習/sample/sample.jpg')
img = cv2.resize(img,dsize=(224,224))
img = img.astype('float32')
img /= 255.0
img = img[None, ...]

# 追加箇所
result = model.predict(img)
pred = result.argmax()

img = cv2.imread('/content/drive/MyDrive/Colab Notebooks/機械学習/sample/sample.jpg')
cv2.imwrite('/content/drive/MyDrive/Colab Notebooks/機械学習/output/' + str(classes[pred])+ '/sample.jpg',img)

参考サイトとしては、
日本語だとtensorflowの公式。「Run in Google colab」を押せばすでにノートブックが用意されています。
https://www.tensorflow.org/tutorials/keras/classification?hl=ja
英語だと
「keras image classification tutorial step by step」や「keras image classification tutorial beginner」などで
検索してみてください。わかりやすい動画やサイトがたくさん出てきます。
英語がわからなくてもコピペや翻訳サイトで逐次見れば問題ないです。

Comments

[Yukkiy]

そうなんですね… あ、あとせっかくプログラムを書いてくださったのに申し訳ないんですが頑張ってエラーを自分で解決できました！！ 他の方に
```
result = model.predict(img)
pred = result.argmax()
```
これを追加してみ？って言われたので追加していろいろして最終的に
```
from keras.models import load_model
import pickle
import cv2
import keras
from sklearn.neighbors import KNeighborsClassifier
from sklearn.metrics import accuracy_score

model = keras.models.load_model('/content/drive/MyDrive/Colab Notebooks/機械学習/model.h5', compile=False)
classes = pickle.load(open('/content/drive/MyDrive/Colab Notebooks/機械学習/classes.sav', 'rb'))

choose = input("どの画像を移動しますか:")
img = cv2.imread("/content/drive/MyDrive/Colab Notebooks/機械学習/sample/" + choose)
img = cv2.resize(img,dsize=(224,224))
img = img.astype('float32')
img /= 255.0
img = img[None, ...]

result = model.predict(img)
pred = result.argmax()

names = ['かしゆか','あ〜ちゃん','のっち']
name = names[pred]

img = cv2.imread("/content/drive/MyDrive/Colab Notebooks/機械学習/sample/" + choose)
cv2.imwrite('/content/drive/MyDrive/Colab Notebooks/機械学習/output/' + name + "/" + choose ,img)
print(choose + "を次のファイルへコピーしました。" + '/content/drive/MyDrive/Colab Notebooks/機械学習/output/' + name)
```
このようにしました！
親切に参考サイトまで教えてくださりありがとうございます。本当にEast_sanさんには感謝、感謝で頭が上がりません。また何か僕が分からなくなったらぜひ教えてください！！