前提・実現したいこと
プログラムをほぼ初めて行う学生です。
自作のデータセットを用いて、tensorflowで機械学習(転移学習)を行いたいです。
行うのは、画像の自動判別です。
その際にGrad-CAMなども利用して画像のどこを認識して学習しているかを可視化したいと思っています。
公式のチュートリアルは動かせるのですが、プログラミングに対する知識が薄いため
どのようにして自作のデータセットを読み込ませればよいのかが分かりません。
調べた際に出てくるのはtensorflow1.Xの際のものや公式のデータセットを用いた行ったものの解説が多く、
tensorflow2以降ではどのように行えば学習させられるのか知りたいです。
かなり大まかな内容となってしまい、すみません。
よろしくお願いします。
試したこと
https://github.com/tensorflow/docs/blob/master/site/en/tutorials/images/transfer_learning.ipynb
このチュートリアルの犬と猫の画像部分を自作データセットに置き換えられたらと思っています。
データセットは見様見真似で作成して、ローカル内に保存しています。
チュートリアルを行う際にも、
https://dev.classmethod.jp/articles/introduce-to-transfer-learning-by-tensorflow-for-beginner/#toc-8
こちらのサイト様を参考にして行いました。
また、
https://qiita.com/AkiyoshiOkano/items/72f3e4ba9caf514460ee
こちらの方が公開してくださっているコードをtensorflow2.1でも動くように
エラーが出た場所を書き直していったのですが、
138行目あたりの書き換え中にvar_listを要求?され、
[]内に何を書き込めば良いのか分からず、つまづいてしまいました。
補足情報(FW/ツールのバージョンなど)
anaconda3を利用して環境構築
tensorflow 2.1
python3.6.10
あなたの回答
tips
プレビュー