2017-11-16 16 views
1

テンソルフローオブジェクト検出apiを自分のデータセットに使用していますが、何らかの問題に直面しています。私はGPU Geforce 1080、8 GB GPUメモリ、tensorflow 1.2.1でcentosを使用しています。トレーニングセットでは500枚、テストでは40枚の画像があります。私は次のステップを踏んで、2つの問題があります。 1. LabelImgツールを使用して画像に注釈を付けました 2.作成したtfrecordが成功しました 3. ssd_inception_v2_coco.configが使用されました。私は唯一のパスを修正しました。クラスはなく、最初から訓練しませんでした。私はssd_inception_v2_cocoモデルチェックポイントを使用しました。トレーニング時にテンソルフローロスが減少しない

問題1:ステップ0から3000まで、私の損失は劇的に減少しましたが、その後は5〜6の間で一定にとどまります。どのように私はそれを減らすが、まだ私のモデルは、必要なオブジェクトを検出することができません。 Here is my Tensorborad samples でも私はdiffentモデルなどを試しました。 eval.pyを実行するために私ができる文書によるが、次のエラーを取得: 警告:ルート:次のクラスは何のグランドトゥルース例がない:0 1と2

問題2の間にいくつかのステップロス滞在定数後faster_rcnn_inception_resnet_v2_atrous_cocoそのプログラムが終了した後。 train.pyとeval.pyを同時に実行しようとすると、同じエラーが発生します。 私に提案をお願いします。私はテンソルフローの初心者が必要な提案です。

答えて

1

あなたがTensorboardで見ている損失曲線はかなり正常です。最初は、損失は非常に急速に低下しますが、時間の経過とともに一見「底を打ち」ます。トレーニングは遅いプロセスです。反復回数を増やすと、時間の経過とともに安定した低下が見られるはずです。

関連する問題