リンク内容
こちらの質問の続きのようなものなのですが
現在
リンク内容
こちらのサイトを参考にさせていただきながらTensorFlowを用いてAIに画像分類の機械学習を行い、それをAndroid StudioでAndroidアプリにしようと挑戦しています。
最上部のリンク内容の質問のおかげでMy DriveへのマウントとMy Drive内にあるimagesと言う名前のフォルダ内の画像データの読み込みはうまく進めることができました。
しかし、参考にさせていただいたQiitaの記事の
「Android Studioでの実装を考慮してモデルの形を微修正してモデルをトレーニング」と言う部分が上手く実行できず困っています
記事の方では
「ソースコードのこの部分
このあと以下のコードセルまでを順次実行するとモデルのトレーニングができる」
history = model.fit_generator( train_data_gen, steps_per_epoch=total_train // batch_size, epochs=epochs, validation_data=val_data_gen, validation_steps=total_val // batch_size )
とあるのですが、これをどうすれば良いのかがわかりません。
リンク内容
上記リンク先で実行しているのですが
epochs = 25 steps_per_epoch = train_generator.n // batch_size validation_steps = validation_generator.n // batch_size history = model.fit_generator(train_generator, steps_per_epoch = steps_per_epoch, epochs=epochs, workers=4, validation_data=validation_generator, validation_steps=validation_steps)
この部分を丸ごと上記の
history = model.fit_generator( train_data_gen, steps_per_epoch=total_train // batch_size, epochs=epochs, validation_data=val_data_gen, validation_steps=total_val // batch_size )
に書き換えているのですが
--------------------------------------------------------------------------- NameError Traceback (most recent call last) <ipython-input-16-01c6f78f4d4f> in <module>() 1 history = model.fit_generator( ----> 2 train_data_gen, 3 steps_per_epoch=total_train // batch_size, 4 epochs=epochs, 5 validation_data=val_data_gen, NameError: name 'train_data_gen' is not defined
このようなエラーが出てしまっています。
base_dir = '/content/drive保存したフォルダのパス'
を実行したのち
train_dir = os.path.join(base_dir, 'train') validation_dir = os.path.join(base_dir, 'val')
を実行すると
Total training L1 images: 152 Total training L4 images: 246 Total training L6 images: 150 Total validation L1 images: 60 Total validation L4 images: 60 Total validation L6 images: 60 /content/drive/My Drive/Gcolab/images/absLeveL/test/gray/train /content/drive/My Drive/Gcolab/images/absLeveL/test/gray/val
のように結果が返ってきているのですが
私の場合下記画像のように実行しても何も返ってきていない状態になっています
最初のエラーと一緒に解決策がわかる方いらっしゃいましたら教えていただけると幸いです。
回答1件
あなたの回答
tips
プレビュー