問題タブ [tensorflowjs-converter]
For questions regarding programming in ECMAScript (JavaScript/JS) and its various dialects/implementations (excluding ActionScript). Note JavaScript is NOT the same as Java! Please include all relevant tags on your question; e.g., [node.js], [jquery], [json], [reactjs], [angular], [ember.js], [vue.js], [typescript], [svelte], etc.
tensorflow - 間違った予測を行う画像分類グラフ モデル
make_image_classifier python スクリプトを使用して、新しい画像セットで mobilenetv2 を再トレーニングしています。私の最終目標は、ブラウザーの tfjs で予測を行うことです。
これはまさに私がやっていることです:
ステップ 1: モデルを再トレーニングする
ステップ 2: tensorflowjs_converter を使用して、tf で保存されたモデルをグラフ モデルに変換する
ステップ 3: 新しいモデルをブラウザーにロードし、画像入力を前処理して、モデルに予測を依頼する
無効な結果が得られます。初期モデルによる予測を確認しましたが、それらは正しいので、変換が適切に行われていないか、初期スクリプトと同じ方法で画像を前処理していないと考えています。
ヘルプ。
PS: コンバーターを実行すると、次のメッセージが表示されます。私が経験していることに直接関係があるかどうかはわかりません。
tensorflow/core/graph/graph_constructor.cc:750 ノード 'StatefulPartitionedCall' には 71 個の出力がありますが、_output_shapes 属性は 605 個の出力の形状を指定します。出力形状が不正確になる場合があります。