前提
Google colaboratryを使用して機械学習を勉強してました。機械学習は始めたてなこともあり、あるサイトを参考にしながらプログラムを書いていました。まだまだ分からないことも多いためエラーを解決することができなかったのでどなたか解決策を教えて頂けると幸いです…
実現したいこと
画像を認識しフォルダー分けをする
発生している問題・エラーメッセージ
NameError: name 'pred' is not defined
該当のソースコード
Python
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.0img = 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
0 コメント