Closed Kanazawanaoaki closed 4 years ago
同じデータセットで学習をしてTensorBoardで可視化をしてみたが、以下のまま変化しなかったのでよくわからなかった。
Potato 0.84のモデルは、JpegとSegmentationClassだけtrain_06をエラーがでないように消したので、そのせいでIndexがズレて上手く学習出来ていない可能性がある。 画像とラベルの対応が上手くなされていなかったとしたら左上にPotato 0.84を出し続けるのもあり得るかも。
もう一度、完全にtrain_06関連のものを消したデータセットで学習を行ったが今度は何も表示されないモデルになった。(どの物体に対しても信頼度が低すぎて上手く行かないのかもしれない。)
何も表示されないと思っていたが、rqtで閾値を変えると0.50のバウンディングボックスが沢山でていた。やはり上手く学習が出来ていないのだろうか。
使用したのは20200515-235908-one_more_dataset
のモデル
使えるデータセットだけを使って、
python3 ./labelme2voc.py data_annotated data_dataset_voc --labels labels.txt
からやりなおしたデータセットを使って学習をさせると若干上手くいっているのかなという結果が出るようになった。使用したモデルは20200516-121940-local_dataset
https://github.com/Kanazawanaoaki/cook_eyes/issues/4 や https://github.com/Kanazawanaoaki/cook_eyes/issues/3 の問題が関係したと思われる。
うまく学習ができるようになったのでcloseする。
自作データセットでSSDの学習を試してみたが、左上にPotatoが表示されるだけのものになってしまった... 使ったのは、
20200515-214217-one_more_dataset
でtrainに使ったjpg画像をパブリッシュして、認識を試してみると以下のような結果になった。 左上にじゃがいもがくる画像が多かったせいか、ほぼ固定位置にpotato 0.84が表示されるか、何も表示されないかになってしまった。