
EfficientNetはクラス分類タスクで精度が良く、しかもパラメータも少なく計算時間が短い。
しかし、デフォルトのinput_shapeが公式ドキュメントに載っておらず、わからなかったので調べてみた。
EfficientNetB0 – (224, 224, 3)
EfficientNetB1 – (240, 240, 3)
EfficientNetB2 – (260, 260, 3)
EfficientNetB3 – (300, 300, 3)
EfficientNetB4 – (380, 380, 3)
EfficientNetB5 – (456, 456, 3)
EfficientNetB6 – (528, 528, 3)
EfficientNetB7 – (600, 600, 3)
参考

Keras documentation: EfficientNet B0 to B7
Keras documentation
efficientnet/efficientnet/model.py at master · qubvel/efficientnet
Implementation of EfficientNet model. Keras and TensorFlow Keras. - qubvel/efficientnet
コメント