質問をすることでしか得られない、回答やアドバイスがある。

15分調べてもわからないことは、質問しよう!

新規登録して質問してみよう
ただいま回答率
85.50%
Keras

Kerasは、TheanoやTensorFlow/CNTK対応のラッパーライブラリです。DeepLearningの数学的部分を短いコードでネットワークとして表現することが可能。DeepLearningの最新手法を迅速に試すことができます。

Python

Pythonは、コードの読みやすさが特徴的なプログラミング言語の1つです。 強い型付け、動的型付けに対応しており、後方互換性がないバージョン2系とバージョン3系が使用されています。 商用製品の開発にも無料で使用でき、OSだけでなく仮想環境にも対応。Unicodeによる文字列操作をサポートしているため、日本語処理も標準で可能です。

Q&A

解決済

1回答

2098閲覧

VGG16を用いた転移学習でdecode_predictions関数で結果を表示させたい

yoshi1300

総合スコア3

Keras

Kerasは、TheanoやTensorFlow/CNTK対応のラッパーライブラリです。DeepLearningの数学的部分を短いコードでネットワークとして表現することが可能。DeepLearningの最新手法を迅速に試すことができます。

Python

Pythonは、コードの読みやすさが特徴的なプログラミング言語の1つです。 強い型付け、動的型付けに対応しており、後方互換性がないバージョン2系とバージョン3系が使用されています。 商用製品の開発にも無料で使用でき、OSだけでなく仮想環境にも対応。Unicodeによる文字列操作をサポートしているため、日本語処理も標準で可能です。

0グッド

0クリップ

投稿2019/08/14 08:58

前提・実現したいこと

VGG16を用いた転移学習を行い、学習させた画像である可能性が最も高い場合そのファイル名を表示させたいです。現在はdecode_predictions関数を使って表示を試みているのですが下記のようにエラーが出てきています。どのように行うべきなのでしょうか。ご指導よろしくお願いいたします。

発生している問題・エラーメッセージ

`decode_predictions` expects a batch of predictions (i.e. a 2D array of shape (samples, 1000)). Found array with shape: (1, 5)

気になる質問をクリップする

クリップした質問は、後からいつでもMYページで確認できます。

またクリップした質問に回答があった際、通知やメールを受け取ることができます。

バッドをするには、ログインかつ

こちらの条件を満たす必要があります。

guest

回答1

0

ベストアンサー

decode_predictions は ImageNet の1000クラス分類問題の学習モデルを利用する場合にのみ利用できる関数なので、転移学習でクラスの種類を変更した場合には利用できません。

5クラス分類問題なのであれば、モデルの出力は (バッチ数, 5) の2次元配列なので、その配列に対して、np.argmax(ret, axis=1) をとれば、各バッチごとの予測ラベルに変換できます。

python

1import numpy as np 2 3class_names = ["a", "b", "c", "d", "e"] 4 5predictions = np.random.randn(10, 5) # predict() の返り値の代わり 6 7pred_labels = np.argmax(predictions, axis=-1) 8print(pred_labels) # [2 0 0 2 2 2 4 2 3 3] 9 10# ラベル名に置き換えた場合 11pred_label_names = [class_names[x] for x in pred_labels] 12print(pred_label_names) # ['d', 'e', 'b', 'a', 'c', 'c', 'a', 'e', 'c', 'c']

投稿2019/08/14 09:17

tiitoi

総合スコア21956

バッドをするには、ログインかつ

こちらの条件を満たす必要があります。

yoshi1300

2019/08/19 06:45

大変参考になりました。ありがとうございました。
guest

あなたの回答

tips

太字

斜体

打ち消し線

見出し

引用テキストの挿入

コードの挿入

リンクの挿入

リストの挿入

番号リストの挿入

表の挿入

水平線の挿入

プレビュー

15分調べてもわからないことは
teratailで質問しよう!

ただいまの回答率
85.50%

質問をまとめることで
思考を整理して素早く解決

テンプレート機能で
簡単に質問をまとめる

質問する

関連した質問