TensorFlowにEfficientNetV2が実装された。
Module: tf.keras.applications.efficientnet_v2 | TensorFlow v2.16.1
DO NOT EDIT.
www.tensorflow.org
しかし、デフォルトのinput_shapeはNoneとなっており、学習時にリサイズさせる画像サイズがわからない。
TensorFlow Hubにはサイズが載っていた。
Google | efficientnet_v2 | Kaggle
Collection of EfficientNet V2 models for image classification and feature extraction trained on Imagenet-1k (ILSVRC-2012...
www.kaggle.com
| V2 S | 384 |
| V2 M | 480 |
| V2 L | 480 |
| V2 B0 | 224 |
| V2 B1 | 240 |
| V2 B2 | 260 |
| V2 B3 | 300 |
B0~B3はV1と同じサイズみたい。
ダブルチェックの意味も兼ねて、EfficientNetV2のソースコードも見てみる。
keras/keras/applications/efficientnet_v2.py at v2.9.0 · keras-team/keras
Deep Learning for humans. Contribute to keras-team/keras development by creating an account on GitHub.
github.com
ちゃんとdefault_sizeというパラメータが用意されていた。
ブラウザでページ内検索していくと、各モデルでTensorFlow Hubに載っていたサイズと一致していた。
これでFine Tuningの画像サイズを決めることが出来た。
コメント