2017-02-02 6 views
0

ケラスを使用して単一のオブジェクトを認識するシステムを作りたいと思います。私の場合、私は車輪を検出します。 1つのオブジェクトのために私のシステムを訓練するにはどうすればいいですか?私は猫や犬を使う前に分類作業をしましたが、今はまったく別の仕事です。私はまだクラス0 =車輪、クラス= 1の非車輪(何かのランダムな画像)で "分類"していますか?この問題で次の手順を実行するにはどうすればよいですか? 1)1つのオブジェクトの列車システム 2)オブジェクトの検出(スライドウィンドウまたはヒートマップ)単一オブジェクト検出ケラス

答えて

1

タスクは、いわゆるバイナリ分類です。 最終層にニューロンが1つしかないことを確認してください(例:Sequential model model.add(Dense(1、... other parameters ...))))、binary_crossentropyを損失関数として使用してください。

これが役に立ちます。

+1

ありがとうございました!だから私は2つのクラス、1つは車輪、他の車輪はありませんか?あなたは2つのクラスがあるので、反対の2として最終層に唯一のニューロンがなぜあるのか教えてもらえますか? – Powisss

+1

あなたは正しいです。私はあなたが今あなたの問題をよく理解していると思います。 2つのニューロンをバイナリ分類に使用することもできます。 見てください:http://stats.stackexchange.com/questions/207049/neural-network-for-binary-classification-use-1-or-2-output-neurons あなたは両方を実装し、どのより良いパフォーマンスを発揮します(例:ノートオーバーフィット)。 あなたの選択はあなたのものです。試してみる ;) –

関連する問題