tensorflow.js を利用して、物体検知の転移学習をしたいと考えています。
tfjs-examples / simple-object-detection を参考に、簡単な物体検知は出来ることを確認したのですが、こちらでは1画像につき、一つの矩形情報しか取得できません。
理由として、以下の箇所でモデルを変更しているからだと思われますが、
https://github.com/tensorflow/tfjs-examples/blob/7338c2ea9a38392fc3aa7a1297e651b4b18ea078/simple-object-detection/train.js#L123-L130
どのように修正すれば良いか検討がつかなく、知恵を貸していただければ幸いです。
あなたの回答
tips
プレビュー