Hub with Keras(Model編)

Hub with Keras(Model編)

◎環境:Python3.7.2/TensorFlow 1.13

概要

TensorFlow/KerasのチュートリアルHub with Kerasをやってみて、調べた内容(モデル関連)をまとめておく。

Model

Hub With Kerasでのモデルは、以下のような形となる。

f:id:storikai:20190514210523p:plain
モデル

※補足:出力の値は、一番大きいものが確率が高い

Model(転移学習)

MobileNet v2の最上位層(クラス分類を行う)のないモデルを使用している。

f:id:storikai:20190514210617p:plain
モデル(転移学習)

※補足:MobileNet(feature_extractor_layer)のパラメーターは学習しない(trainable=False)