現在、Deep Learningについての学習をしています。CNNを用いたDeep Learningを学習し、どのように学習用データを作成し学習させ、正答率を出力することまではできるようになりました。しかし、これだけでは面白くないと思い、学習させることにより得た特徴を見つけたら枠で囲む物体検出を行えるようにしようと考え以下のサイトを参考にSSD学習を行おうと考えました。
SSD学習アルゴリズム
また、自作のデータセットを使用したかったため、以下のサイトをもとにデータセットを作成しました。
データセット作成
動かしてみたところ、eporch1が終了した段階で以下のようなエラーが出てしまいました。
output of generator should be a tuple(x, y, sample_weight) or (x,y). Found: None
使用したコードは「SSD学習アルゴリズム」に載っていたtrain.pyをディレクトリを変えて使わせていただきましたました。
一応調べてはみたのですが、原因についてはわからない状況です。改善方法について教えていただきたいです。
kerasはバージョン1.2.2を使用しています。回答よろしくお願いします。
あなたの回答
tips
プレビュー