EfficientNet模型的完整细节
极市平台
共 2604字,需浏览 6分钟
·
2020-09-15 09:34
极市导读
本文介绍了一种高效的网络模型EfficientNet,并分析了 EfficientNet B0 至B7的网络结构之间的差异。
共同之处
!pip install tf-nightly-gpu
import tensorflow as tf
IMG_SHAPE = (224, 224, 3)
model0 = tf.keras.applications.EfficientNetB0(input_shape=IMG_SHAPE, include_top=False, weights="imagenet")
tf.keras.utils.plot_model(model0) # to draw and visualize
model0.summary() # to see the list of layers and parameters
模块1 — 这是子block的起点。 模块2 — 此模块用于除第一个模块外的所有7个主要模块的第一个子block的起点。 模块3 — 它作为跳跃连接到所有的子block。 模块4 — 用于将跳跃连接合并到第一个子block中。 模块5 — 每个子block都以跳跃连接的方式连接到之前的子block,并使用此模块进行组合。
子block1 — 它仅用于第一个block中的第一个子block。 子block2 — 它用作所有其他block中的第一个子block。 子block3 — 用于所有block中除第一个外的任何子block。
模型结构
EfficientNet-B0
EfficientNet-B1
EfficientNet-B2
EfficientNet-B3
EfficientNet-B4
EfficientNet-B5
EfficientNet-B6
EfficientNet-B7
推荐阅读
评论