最終的なノートブック

Colaboratory 上でノートブックを作る

から新しいノートブックを作ります。

を選択すると次のようなポップアップが表示されるので、ハードウェアアクセラレート のドロップダウンメニューから GPU を選びましょう。

!pip list | grep tensorflow

上記を実行すると、バージョンが 1.15.0 であることがわかります。

!pip install -U tensorflow

上記を実行すると次のようなエラーが出ますが、今回の範囲内では問題ありませんでした。

もう一度 pip list でバージョンを確認しておきましょう。

たったこれだけです。

import tensorflow as tf

resnet = tf.keras.applications.ResNet50V2(include_top=True, weights="imagenet")

入力と出力の形式を確認します。
入力画像は 224x224 であることがわかります。

resnet.input_shape

resnet.output_shape

左ペインの ファイル から画像をアップロードします。
ここでは、dog.jpg という名前です。

次に、画像を読み込み、ResNetの入力用に変形します。

import numpy as np
from PIL import Image

img = Image.open("dog.jpg")
input = np.array(img.convert("RGB").resize((224, 224)))
input.shape

前処理をする必要があるみたいです。
入力した画像の数xラベル数 になっていることがわかります。

preprocessed = tf.keras.applications.resnet_v2.preprocess_input(np.array([input]))
output = resnet.predict(preprocessed)
output.shape

結果をデコードします。

result = tf.keras.applications.resnet_v2.decode_predictions(output, top=10)[0]
result

入力画像と比較すると、概ね納得できる結果でした。