qubvel / segmentation_models

Segmentation models with pretrained backbones. Keras and TensorFlow Keras.

Geek Repo:Geek Repo

Github PK Tool:Github PK Tool

Are layers missing in EfficientNet backbone (block 5)?

rdmato33 opened this issue · comments

I recently compared your EfficientNet-B0 backbone with the state-of-art architecture and detected 4 layers were missing in block 5. Only subblock 5a, 5b and 5c appear when ploting the model (no "skips" are implemented). Where 5d, 5e, 5f and 5g subblocks are (MBConv6 5X5) since they appear in the state-of-art architecture (https://ai.googleblog.com/2019/05/efficientnet-improving-accuracy-and.html)?

Here the code used to plot it:

from keras.utils.vis_utils import plot_model
plot_model(model, to_file='model_plot.png', show_shapes=True, show_layer_names=True)

Thanks a lot.