YOLOでの学習状況の質問

15 views (last 30 days)
HY
HY on 14 Jul 2020
Commented: Kenta on 16 Jul 2020
YOLOを学習させた際に、学習の結果が以下の画像のようになりました。
学習オプションは、
trainingOptions('sgdm', ...
'MiniBatchSize', 64, ....
'InitialLearnRate',1e-3, ...
'MaxEpochs',30,...
'CheckpointPath', tempdir, ...
'Shuffle','never');
となっています。
この後、テストデータで検証を試みるものの、全く認識ができません。謎なことに、学習データで検証してもbboxesが空になってしまいます。
(ちなみに、学習データ数は5000枚以上になっています。)
ここで質問ですが、画像のように学習状況が適切にできているように見えても、実際は学習できない場合があるのでしょうか。
まだまだ不勉強なため、大雑把な質問となってしまい申し訳ございませんが、何か教えていただければ幸いです。

Accepted Answer

Kenta
Kenta on 14 Jul 2020
Edited: Kenta on 14 Jul 2020
こんにちは、matlabのバージョンはいくつでされていますか?2019aなどの少し前のものでされていますか?もし、そうであれば、最新の2020aにアップデートしてやってみればよいかもしれません。もしくは、単に、すべてのセルに対して物体がない、というような判断を下すように学習が進んでしまっているのかもしれないな、とも思ったのですが、それにしては十分Lossが小さくなっている気もします。
  12 Comments
HY
HY on 16 Jul 2020
はい、明日アップデートできそうなので、試してみます。
みなさま、誠にありがとうございました。
Kenta
Kenta on 16 Jul 2020
はい、うまくいくといいですね!

Sign in to comment.

More Answers (0)

Community Treasure Hunt

Find the treasures in MATLAB Central and discover how the community can help you!

Start Hunting!