Kanazawanaoaki / cook_eyes

Implement the eyes of the cook
1 stars 2 forks source link

Potato 0.84が左上に表示されるだけになってしまった #1

Closed Kanazawanaoaki closed 4 years ago

Kanazawanaoaki commented 4 years ago

自作データセットでSSDの学習を試してみたが、左上にPotatoが表示されるだけのものになってしまった... 使ったのは、20200515-214217-one_more_dataset

$ rosrun jsk_perception image_publisher.py _file_name:=FILE_PATH

でtrainに使ったjpg画像をパブリッシュして、認識を試してみると以下のような結果になった。 左上にじゃがいもがくる画像が多かったせいか、ほぼ固定位置にpotato 0.84が表示されるか、何も表示されないかになってしまった。

Screenshot from 2020-05-15 22-26-17 Screenshot from 2020-05-15 22-26-27 Screenshot from 2020-05-15 22-26-49 Screenshot from 2020-05-15 22-27-01 Screenshot from 2020-05-15 22-27-28 Screenshot from 2020-05-15 22-27-39 Screenshot from 2020-05-15 22-27-50

Kanazawanaoaki commented 4 years ago

同じデータセットで学習をしてTensorBoardで可視化をしてみたが、以下のまま変化しなかったのでよくわからなかった。
Screenshot from 2020-05-15 23-49-16 Screenshot from 2020-05-15 23-49-14 Screenshot from 2020-05-15 23-42-44

Kanazawanaoaki commented 4 years ago

Potato 0.84のモデルは、JpegとSegmentationClassだけtrain_06をエラーがでないように消したので、そのせいでIndexがズレて上手く学習出来ていない可能性がある。 画像とラベルの対応が上手くなされていなかったとしたら左上にPotato 0.84を出し続けるのもあり得るかも。

もう一度、完全にtrain_06関連のものを消したデータセットで学習を行ったが今度は何も表示されないモデルになった。(どの物体に対しても信頼度が低すぎて上手く行かないのかもしれない。)

Kanazawanaoaki commented 4 years ago

何も表示されないと思っていたが、rqtで閾値を変えると0.50のバウンディングボックスが沢山でていた。やはり上手く学習が出来ていないのだろうか。 Screenshot from 2020-05-16 01-07-22 使用したのは20200515-235908-one_more_datasetのモデル

Kanazawanaoaki commented 4 years ago

使えるデータセットだけを使って、

python3 ./labelme2voc.py data_annotated data_dataset_voc --labels labels.txt

からやりなおしたデータセットを使って学習をさせると若干上手くいっているのかなという結果が出るようになった。使用したモデルは20200516-121940-local_dataset Screenshot from 2020-05-16 13-10-14 Screenshot from 2020-05-16 13-10-54

Kanazawanaoaki commented 4 years ago

https://github.com/Kanazawanaoaki/cook_eyes/issues/4https://github.com/Kanazawanaoaki/cook_eyes/issues/3 の問題が関係したと思われる。
うまく学習ができるようになったのでcloseする。