問題タブ [imagenet]

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.

0 投票する
2 に答える
1960 参照

python - TensorFlow/Keras - global_average_pooling2d_1_input の形状は (1、1、2048) であると予想されていましたが、形状 (7、7、2048) の配列を取得しました

私は TensorFlow と画像分類にかなり慣れていないため、重要な知識が不足している可能性があり、おそらくこの問題に直面している理由です。

ResNet50ライブラリを使用して犬種の画像分類を目的として TensorFlow でモデルを構築し、ImageNetさまざまな犬種を検出できるニューラル ネットワークのトレーニングに成功しました。

私は今、犬のランダムな画像をモデルに渡して、犬の品種が何であるかについての出力を吐き出したいところです。ただし、この関数を実行すると、行を実行しようとするとdog_breed_predictor("<file path to image>")エラーが発生し、これを回避する方法がわかりません。expected global_average_pooling2d_1_input to have shape (1, 1, 2048) but got array with shape (7, 7, 2048)Resnet50_model.predict(bottleneck_feature)

これがコードです。問題に関連していると思われるものはすべて提供しました。

関数に入力している画像は、モデルのトレーニングに使用されたものと同じデータセットからのものです。モデルが意図したとおりに機能しているかどうかを確認したかったため、このエラーによりさらに混乱します。私は何が間違っているのでしょうか?