ubuntu虚拟环境virtualenv, tf-nightly, efficientnet安装

29 篇文章 2 订阅
10 篇文章 1 订阅

因要用到efficientnet模型,现在这个模型只存在在tf-nightly版本中, 又不想破坏主环境,因此采用虚拟环境virtualenv来安装,而不是采用conda create.
sudo apt-get install python-virtualenv
virtualenv -p /usr/bin/python3 tf2
进入env
source ~/tf2/bin/activate
查看虚拟环境下安装的所有的包 
pip list
查看虚拟环境重通过pip安装的包
pip freeze
pip install tf-nightly-gpu
这时的GPU仍有可能与主环境的CUDA不匹配而不能应用,如果要用GPU,需要重新安装CUDA
验证,keras已包含在tensorflow中,不必单独安装
python
import tensorflow
from tensorflow import keras
安装依赖项
pip install graphviz
pip install pydot
运行程序:

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

退出env
deactivate

在这里插入图片描述

在这里插入图片描述
Model: “efficientnetb0”


Layer (type) Output Shape Param # Connected to

input_1 (InputLayer) [(None, 224, 224, 3) 0


rescaling (Rescaling) (None, 224, 224, 3) 0 input_1[0][0]


normalization (Normalization) (None, 224, 224, 3) 7 rescaling[0][0]


stem_conv_pad (ZeroPadding2D) (None, 225, 225, 3) 0 normalization[0][0]


stem_conv (Conv2D) (None, 112, 112, 32) 864 stem_conv_pad[0][0]


stem_bn (BatchNormalization) (None, 112, 112, 32) 128 stem_conv[0][0]


stem_activation (Activation) (None, 112, 112, 32) 0 stem_bn[0][0]


block1a_dwconv (DepthwiseConv2D (None, 112, 112, 32) 288 stem_activation[0][0]


block1a_bn (BatchNormalization) (None, 112, 112, 32) 128 block1a_dwconv[0][0]


block1a_activation (Activation) (None, 112, 112, 32) 0 block1a_bn[0][0]


block1a_se_squeeze (GlobalAvera (None, 32) 0 block1a_activation[0][0]


block1a_se_reshape (Reshape) (None, 1, 1, 32) 0 block1a_se_squeeze[0][0]


block1a_se_reduce (Conv2D) (None, 1, 1, 8) 264 block1a_se_reshape[0][0]


block1a_se_expand (Conv2D) (None, 1, 1, 32) 288 block1a_se_reduce[0][0]


block1a_se_excite (Multiply) (None, 112, 112, 32) 0 block1a_activation[0][0]
block1a_se_expand[0][0]


block1a_project_conv (Conv2D) (None, 112, 112, 16) 512 block1a_se_excite[0][0]


block1a_project_bn (BatchNormal (None, 112, 112, 16) 64 block1a_project_conv[0][0]


block2a_expand_conv (Conv2D) (None, 112, 112, 96) 1536 block1a_project_bn[0][0]


block2a_expand_bn (BatchNormali (None, 112, 112, 96) 384 block2a_expand_conv[0][0]


block2a_expand_activation (Acti (None, 112, 112, 96) 0 block2a_expand_bn[0][0]


block2a_dwconv_pad (ZeroPadding (None, 113, 113, 96) 0 block2a_expand_activation[0][0]


block2a_dwconv (DepthwiseConv2D (None, 56, 56, 96) 864 block2a_dwconv_pad[0][0]


block2a_bn (BatchNormalization) (None, 56, 56, 96) 384 block2a_dwconv[0][0]


block2a_activation (Activation) (None, 56, 56, 96) 0 block2a_bn[0][0]


block2a_se_squeeze (GlobalAvera (None, 96) 0 block2a_activation[0][0]


block2a_se_reshape (Reshape) (None, 1, 1, 96) 0 block2a_se_squeeze[0][0]


block2a_se_reduce (Conv2D) (None, 1, 1, 4) 388 block2a_se_reshape[0][0]


block2a_se_expand (Conv2D) (None, 1, 1, 96) 480 block2a_se_reduce[0][0]


block2a_se_excite (Multiply) (None, 56, 56, 96) 0 block2a_activation[0][0]
block2a_se_expand[0][0]


block2a_project_conv (Conv2D) (None, 56, 56, 24) 2304 block2a_se_excite[0][0]


block2a_project_bn (BatchNormal (None, 56, 56, 24) 96 block2a_project_conv[0][0]


block2b_expand_conv (Conv2D) (None, 56, 56, 144) 3456 block2a_project_bn[0][0]


block2b_expand_bn (BatchNormali (None, 56, 56, 144) 576 block2b_expand_conv[0][0]


block2b_expand_activation (Acti (None, 56, 56, 144) 0 block2b_expand_bn[0][0]


block2b_dwconv (DepthwiseConv2D (None, 56, 56, 144) 1296 block2b_expand_activation[0][0]


block2b_bn (BatchNormalization) (None, 56, 56, 144) 576 block2b_dwconv[0][0]


block2b_activation (Activation) (None, 56, 56, 144) 0 block2b_bn[0][0]


block2b_se_squeeze (GlobalAvera (None, 144) 0 block2b_activation[0][0]


block2b_se_reshape (Reshape) (None, 1, 1, 144) 0 block2b_se_squeeze[0][0]


block2b_se_reduce (Conv2D) (None, 1, 1, 6) 870 block2b_se_reshape[0][0]


block2b_se_expand (Conv2D) (None, 1, 1, 144) 1008 block2b_se_reduce[0][0]


block2b_se_excite (Multiply) (None, 56, 56, 144) 0 block2b_activation[0][0]
block2b_se_expand[0][0]


block2b_project_conv (Conv2D) (None, 56, 56, 24) 3456 block2b_se_excite[0][0]


block2b_project_bn (BatchNormal (None, 56, 56, 24) 96 block2b_project_conv[0][0]


block2b_drop (Dropout) (None, 56, 56, 24) 0 block2b_project_bn[0][0]


block2b_add (Add) (None, 56, 56, 24) 0 block2b_drop[0][0]
block2a_project_bn[0][0]


block3a_expand_conv (Conv2D) (None, 56, 56, 144) 3456 block2b_add[0][0]


block3a_expand_bn (BatchNormali (None, 56, 56, 144) 576 block3a_expand_conv[0][0]


block3a_expand_activation (Acti (None, 56, 56, 144) 0 block3a_expand_bn[0][0]


block3a_dwconv_pad (ZeroPadding (None, 59, 59, 144) 0 block3a_expand_activation[0][0]


block3a_dwconv (DepthwiseConv2D (None, 28, 28, 144) 3600 block3a_dwconv_pad[0][0]


block3a_bn (BatchNormalization) (None, 28, 28, 144) 576 block3a_dwconv[0][0]


block3a_activation (Activation) (None, 28, 28, 144) 0 block3a_bn[0][0]


block3a_se_squeeze (GlobalAvera (None, 144) 0 block3a_activation[0][0]


block3a_se_reshape (Reshape) (None, 1, 1, 144) 0 block3a_se_squeeze[0][0]


block3a_se_reduce (Conv2D) (None, 1, 1, 6) 870 block3a_se_reshape[0][0]


block3a_se_expand (Conv2D) (None, 1, 1, 144) 1008 block3a_se_reduce[0][0]


block3a_se_excite (Multiply) (None, 28, 28, 144) 0 block3a_activation[0][0]
block3a_se_expand[0][0]


block3a_project_conv (Conv2D) (None, 28, 28, 40) 5760 block3a_se_excite[0][0]


block3a_project_bn (BatchNormal (None, 28, 28, 40) 160 block3a_project_conv[0][0]


block3b_expand_conv (Conv2D) (None, 28, 28, 240) 9600 block3a_project_bn[0][0]


block3b_expand_bn (BatchNormali (None, 28, 28, 240) 960 block3b_expand_conv[0][0]


block3b_expand_activation (Acti (None, 28, 28, 240) 0 block3b_expand_bn[0][0]


block3b_dwconv (DepthwiseConv2D (None, 28, 28, 240) 6000 block3b_expand_activation[0][0]


block3b_bn (BatchNormalization) (None, 28, 28, 240) 960 block3b_dwconv[0][0]


block3b_activation (Activation) (None, 28, 28, 240) 0 block3b_bn[0][0]


block3b_se_squeeze (GlobalAvera (None, 240) 0 block3b_activation[0][0]


block3b_se_reshape (Reshape) (None, 1, 1, 240) 0 block3b_se_squeeze[0][0]


block3b_se_reduce (Conv2D) (None, 1, 1, 10) 2410 block3b_se_reshape[0][0]


block3b_se_expand (Conv2D) (None, 1, 1, 240) 2640 block3b_se_reduce[0][0]


block3b_se_excite (Multiply) (None, 28, 28, 240) 0 block3b_activation[0][0]
block3b_se_expand[0][0]


block3b_project_conv (Conv2D) (None, 28, 28, 40) 9600 block3b_se_excite[0][0]


block3b_project_bn (BatchNormal (None, 28, 28, 40) 160 block3b_project_conv[0][0]


block3b_drop (Dropout) (None, 28, 28, 40) 0 block3b_project_bn[0][0]


block3b_add (Add) (None, 28, 28, 40) 0 block3b_drop[0][0]
block3a_project_bn[0][0]


block4a_expand_conv (Conv2D) (None, 28, 28, 240) 9600 block3b_add[0][0]


block4a_expand_bn (BatchNormali (None, 28, 28, 240) 960 block4a_expand_conv[0][0]


block4a_expand_activation (Acti (None, 28, 28, 240) 0 block4a_expand_bn[0][0]


block4a_dwconv_pad (ZeroPadding (None, 29, 29, 240) 0 block4a_expand_activation[0][0]


block4a_dwconv (DepthwiseConv2D (None, 14, 14, 240) 2160 block4a_dwconv_pad[0][0]


block4a_bn (BatchNormalization) (None, 14, 14, 240) 960 block4a_dwconv[0][0]


block4a_activation (Activation) (None, 14, 14, 240) 0 block4a_bn[0][0]


block4a_se_squeeze (GlobalAvera (None, 240) 0 block4a_activation[0][0]


block4a_se_reshape (Reshape) (None, 1, 1, 240) 0 block4a_se_squeeze[0][0]


block4a_se_reduce (Conv2D) (None, 1, 1, 10) 2410 block4a_se_reshape[0][0]


block4a_se_expand (Conv2D) (None, 1, 1, 240) 2640 block4a_se_reduce[0][0]


block4a_se_excite (Multiply) (None, 14, 14, 240) 0 block4a_activation[0][0]
block4a_se_expand[0][0]


block4a_project_conv (Conv2D) (None, 14, 14, 80) 19200 block4a_se_excite[0][0]


block4a_project_bn (BatchNormal (None, 14, 14, 80) 320 block4a_project_conv[0][0]


block4b_expand_conv (Conv2D) (None, 14, 14, 480) 38400 block4a_project_bn[0][0]


block4b_expand_bn (BatchNormali (None, 14, 14, 480) 1920 block4b_expand_conv[0][0]


block4b_expand_activation (Acti (None, 14, 14, 480) 0 block4b_expand_bn[0][0]


block4b_dwconv (DepthwiseConv2D (None, 14, 14, 480) 4320 block4b_expand_activation[0][0]


block4b_bn (BatchNormalization) (None, 14, 14, 480) 1920 block4b_dwconv[0][0]


block4b_activation (Activation) (None, 14, 14, 480) 0 block4b_bn[0][0]


block4b_se_squeeze (GlobalAvera (None, 480) 0 block4b_activation[0][0]


block4b_se_reshape (Reshape) (None, 1, 1, 480) 0 block4b_se_squeeze[0][0]


block4b_se_reduce (Conv2D) (None, 1, 1, 20) 9620 block4b_se_reshape[0][0]


block4b_se_expand (Conv2D) (None, 1, 1, 480) 10080 block4b_se_reduce[0][0]


block4b_se_excite (Multiply) (None, 14, 14, 480) 0 block4b_activation[0][0]
block4b_se_expand[0][0]


block4b_project_conv (Conv2D) (None, 14, 14, 80) 38400 block4b_se_excite[0][0]


block4b_project_bn (BatchNormal (None, 14, 14, 80) 320 block4b_project_conv[0][0]


block4b_drop (Dropout) (None, 14, 14, 80) 0 block4b_project_bn[0][0]


block4b_add (Add) (None, 14, 14, 80) 0 block4b_drop[0][0]
block4a_project_bn[0][0]


block4c_expand_conv (Conv2D) (None, 14, 14, 480) 38400 block4b_add[0][0]


block4c_expand_bn (BatchNormali (None, 14, 14, 480) 1920 block4c_expand_conv[0][0]


block4c_expand_activation (Acti (None, 14, 14, 480) 0 block4c_expand_bn[0][0]


block4c_dwconv (DepthwiseConv2D (None, 14, 14, 480) 4320 block4c_expand_activation[0][0]


block4c_bn (BatchNormalization) (None, 14, 14, 480) 1920 block4c_dwconv[0][0]


block4c_activation (Activation) (None, 14, 14, 480) 0 block4c_bn[0][0]


block4c_se_squeeze (GlobalAvera (None, 480) 0 block4c_activation[0][0]


block4c_se_reshape (Reshape) (None, 1, 1, 480) 0 block4c_se_squeeze[0][0]


block4c_se_reduce (Conv2D) (None, 1, 1, 20) 9620 block4c_se_reshape[0][0]


block4c_se_expand (Conv2D) (None, 1, 1, 480) 10080 block4c_se_reduce[0][0]


block4c_se_excite (Multiply) (None, 14, 14, 480) 0 block4c_activation[0][0]
block4c_se_expand[0][0]


block4c_project_conv (Conv2D) (None, 14, 14, 80) 38400 block4c_se_excite[0][0]


block4c_project_bn (BatchNormal (None, 14, 14, 80) 320 block4c_project_conv[0][0]


block4c_drop (Dropout) (None, 14, 14, 80) 0 block4c_project_bn[0][0]


block4c_add (Add) (None, 14, 14, 80) 0 block4c_drop[0][0]
block4b_add[0][0]


block5a_expand_conv (Conv2D) (None, 14, 14, 480) 38400 block4c_add[0][0]


block5a_expand_bn (BatchNormali (None, 14, 14, 480) 1920 block5a_expand_conv[0][0]


block5a_expand_activation (Acti (None, 14, 14, 480) 0 block5a_expand_bn[0][0]


block5a_dwconv (DepthwiseConv2D (None, 14, 14, 480) 12000 block5a_expand_activation[0][0]


block5a_bn (BatchNormalization) (None, 14, 14, 480) 1920 block5a_dwconv[0][0]


block5a_activation (Activation) (None, 14, 14, 480) 0 block5a_bn[0][0]


block5a_se_squeeze (GlobalAvera (None, 480) 0 block5a_activation[0][0]


block5a_se_reshape (Reshape) (None, 1, 1, 480) 0 block5a_se_squeeze[0][0]


block5a_se_reduce (Conv2D) (None, 1, 1, 20) 9620 block5a_se_reshape[0][0]


block5a_se_expand (Conv2D) (None, 1, 1, 480) 10080 block5a_se_reduce[0][0]


block5a_se_excite (Multiply) (None, 14, 14, 480) 0 block5a_activation[0][0]
block5a_se_expand[0][0]


block5a_project_conv (Conv2D) (None, 14, 14, 112) 53760 block5a_se_excite[0][0]


block5a_project_bn (BatchNormal (None, 14, 14, 112) 448 block5a_project_conv[0][0]


block5b_expand_conv (Conv2D) (None, 14, 14, 672) 75264 block5a_project_bn[0][0]


block5b_expand_bn (BatchNormali (None, 14, 14, 672) 2688 block5b_expand_conv[0][0]


block5b_expand_activation (Acti (None, 14, 14, 672) 0 block5b_expand_bn[0][0]


block5b_dwconv (DepthwiseConv2D (None, 14, 14, 672) 16800 block5b_expand_activation[0][0]


block5b_bn (BatchNormalization) (None, 14, 14, 672) 2688 block5b_dwconv[0][0]


block5b_activation (Activation) (None, 14, 14, 672) 0 block5b_bn[0][0]


block5b_se_squeeze (GlobalAvera (None, 672) 0 block5b_activation[0][0]


block5b_se_reshape (Reshape) (None, 1, 1, 672) 0 block5b_se_squeeze[0][0]


block5b_se_reduce (Conv2D) (None, 1, 1, 28) 18844 block5b_se_reshape[0][0]


block5b_se_expand (Conv2D) (None, 1, 1, 672) 19488 block5b_se_reduce[0][0]


block5b_se_excite (Multiply) (None, 14, 14, 672) 0 block5b_activation[0][0]
block5b_se_expand[0][0]


block5b_project_conv (Conv2D) (None, 14, 14, 112) 75264 block5b_se_excite[0][0]


block5b_project_bn (BatchNormal (None, 14, 14, 112) 448 block5b_project_conv[0][0]


block5b_drop (Dropout) (None, 14, 14, 112) 0 block5b_project_bn[0][0]


block5b_add (Add) (None, 14, 14, 112) 0 block5b_drop[0][0]
block5a_project_bn[0][0]


block5c_expand_conv (Conv2D) (None, 14, 14, 672) 75264 block5b_add[0][0]


block5c_expand_bn (BatchNormali (None, 14, 14, 672) 2688 block5c_expand_conv[0][0]


block5c_expand_activation (Acti (None, 14, 14, 672) 0 block5c_expand_bn[0][0]


block5c_dwconv (DepthwiseConv2D (None, 14, 14, 672) 16800 block5c_expand_activation[0][0]


block5c_bn (BatchNormalization) (None, 14, 14, 672) 2688 block5c_dwconv[0][0]


block5c_activation (Activation) (None, 14, 14, 672) 0 block5c_bn[0][0]


block5c_se_squeeze (GlobalAvera (None, 672) 0 block5c_activation[0][0]


block5c_se_reshape (Reshape) (None, 1, 1, 672) 0 block5c_se_squeeze[0][0]


block5c_se_reduce (Conv2D) (None, 1, 1, 28) 18844 block5c_se_reshape[0][0]


block5c_se_expand (Conv2D) (None, 1, 1, 672) 19488 block5c_se_reduce[0][0]


block5c_se_excite (Multiply) (None, 14, 14, 672) 0 block5c_activation[0][0]
block5c_se_expand[0][0]


block5c_project_conv (Conv2D) (None, 14, 14, 112) 75264 block5c_se_excite[0][0]


block5c_project_bn (BatchNormal (None, 14, 14, 112) 448 block5c_project_conv[0][0]


block5c_drop (Dropout) (None, 14, 14, 112) 0 block5c_project_bn[0][0]


block5c_add (Add) (None, 14, 14, 112) 0 block5c_drop[0][0]
block5b_add[0][0]


block6a_expand_conv (Conv2D) (None, 14, 14, 672) 75264 block5c_add[0][0]


block6a_expand_bn (BatchNormali (None, 14, 14, 672) 2688 block6a_expand_conv[0][0]


block6a_expand_activation (Acti (None, 14, 14, 672) 0 block6a_expand_bn[0][0]


block6a_dwconv_pad (ZeroPadding (None, 17, 17, 672) 0 block6a_expand_activation[0][0]


block6a_dwconv (DepthwiseConv2D (None, 7, 7, 672) 16800 block6a_dwconv_pad[0][0]


block6a_bn (BatchNormalization) (None, 7, 7, 672) 2688 block6a_dwconv[0][0]


block6a_activation (Activation) (None, 7, 7, 672) 0 block6a_bn[0][0]


block6a_se_squeeze (GlobalAvera (None, 672) 0 block6a_activation[0][0]


block6a_se_reshape (Reshape) (None, 1, 1, 672) 0 block6a_se_squeeze[0][0]


block6a_se_reduce (Conv2D) (None, 1, 1, 28) 18844 block6a_se_reshape[0][0]


block6a_se_expand (Conv2D) (None, 1, 1, 672) 19488 block6a_se_reduce[0][0]


block6a_se_excite (Multiply) (None, 7, 7, 672) 0 block6a_activation[0][0]
block6a_se_expand[0][0]


block6a_project_conv (Conv2D) (None, 7, 7, 192) 129024 block6a_se_excite[0][0]


block6a_project_bn (BatchNormal (None, 7, 7, 192) 768 block6a_project_conv[0][0]


block6b_expand_conv (Conv2D) (None, 7, 7, 1152) 221184 block6a_project_bn[0][0]


block6b_expand_bn (BatchNormali (None, 7, 7, 1152) 4608 block6b_expand_conv[0][0]


block6b_expand_activation (Acti (None, 7, 7, 1152) 0 block6b_expand_bn[0][0]


block6b_dwconv (DepthwiseConv2D (None, 7, 7, 1152) 28800 block6b_expand_activation[0][0]


block6b_bn (BatchNormalization) (None, 7, 7, 1152) 4608 block6b_dwconv[0][0]


block6b_activation (Activation) (None, 7, 7, 1152) 0 block6b_bn[0][0]


block6b_se_squeeze (GlobalAvera (None, 1152) 0 block6b_activation[0][0]


block6b_se_reshape (Reshape) (None, 1, 1, 1152) 0 block6b_se_squeeze[0][0]


block6b_se_reduce (Conv2D) (None, 1, 1, 48) 55344 block6b_se_reshape[0][0]


block6b_se_expand (Conv2D) (None, 1, 1, 1152) 56448 block6b_se_reduce[0][0]


block6b_se_excite (Multiply) (None, 7, 7, 1152) 0 block6b_activation[0][0]
block6b_se_expand[0][0]


block6b_project_conv (Conv2D) (None, 7, 7, 192) 221184 block6b_se_excite[0][0]


block6b_project_bn (BatchNormal (None, 7, 7, 192) 768 block6b_project_conv[0][0]


block6b_drop (Dropout) (None, 7, 7, 192) 0 block6b_project_bn[0][0]


block6b_add (Add) (None, 7, 7, 192) 0 block6b_drop[0][0]
block6a_project_bn[0][0]


block6c_expand_conv (Conv2D) (None, 7, 7, 1152) 221184 block6b_add[0][0]


block6c_expand_bn (BatchNormali (None, 7, 7, 1152) 4608 block6c_expand_conv[0][0]


block6c_expand_activation (Acti (None, 7, 7, 1152) 0 block6c_expand_bn[0][0]


block6c_dwconv (DepthwiseConv2D (None, 7, 7, 1152) 28800 block6c_expand_activation[0][0]


block6c_bn (BatchNormalization) (None, 7, 7, 1152) 4608 block6c_dwconv[0][0]


block6c_activation (Activation) (None, 7, 7, 1152) 0 block6c_bn[0][0]


block6c_se_squeeze (GlobalAvera (None, 1152) 0 block6c_activation[0][0]


block6c_se_reshape (Reshape) (None, 1, 1, 1152) 0 block6c_se_squeeze[0][0]


block6c_se_reduce (Conv2D) (None, 1, 1, 48) 55344 block6c_se_reshape[0][0]


block6c_se_expand (Conv2D) (None, 1, 1, 1152) 56448 block6c_se_reduce[0][0]


block6c_se_excite (Multiply) (None, 7, 7, 1152) 0 block6c_activation[0][0]
block6c_se_expand[0][0]


block6c_project_conv (Conv2D) (None, 7, 7, 192) 221184 block6c_se_excite[0][0]


block6c_project_bn (BatchNormal (None, 7, 7, 192) 768 block6c_project_conv[0][0]


block6c_drop (Dropout) (None, 7, 7, 192) 0 block6c_project_bn[0][0]


block6c_add (Add) (None, 7, 7, 192) 0 block6c_drop[0][0]
block6b_add[0][0]


block6d_expand_conv (Conv2D) (None, 7, 7, 1152) 221184 block6c_add[0][0]


block6d_expand_bn (BatchNormali (None, 7, 7, 1152) 4608 block6d_expand_conv[0][0]


block6d_expand_activation (Acti (None, 7, 7, 1152) 0 block6d_expand_bn[0][0]


block6d_dwconv (DepthwiseConv2D (None, 7, 7, 1152) 28800 block6d_expand_activation[0][0]


block6d_bn (BatchNormalization) (None, 7, 7, 1152) 4608 block6d_dwconv[0][0]


block6d_activation (Activation) (None, 7, 7, 1152) 0 block6d_bn[0][0]


block6d_se_squeeze (GlobalAvera (None, 1152) 0 block6d_activation[0][0]


block6d_se_reshape (Reshape) (None, 1, 1, 1152) 0 block6d_se_squeeze[0][0]


block6d_se_reduce (Conv2D) (None, 1, 1, 48) 55344 block6d_se_reshape[0][0]


block6d_se_expand (Conv2D) (None, 1, 1, 1152) 56448 block6d_se_reduce[0][0]


block6d_se_excite (Multiply) (None, 7, 7, 1152) 0 block6d_activation[0][0]
block6d_se_expand[0][0]


block6d_project_conv (Conv2D) (None, 7, 7, 192) 221184 block6d_se_excite[0][0]


block6d_project_bn (BatchNormal (None, 7, 7, 192) 768 block6d_project_conv[0][0]


block6d_drop (Dropout) (None, 7, 7, 192) 0 block6d_project_bn[0][0]


block6d_add (Add) (None, 7, 7, 192) 0 block6d_drop[0][0]
block6c_add[0][0]


block7a_expand_conv (Conv2D) (None, 7, 7, 1152) 221184 block6d_add[0][0]


block7a_expand_bn (BatchNormali (None, 7, 7, 1152) 4608 block7a_expand_conv[0][0]


block7a_expand_activation (Acti (None, 7, 7, 1152) 0 block7a_expand_bn[0][0]


block7a_dwconv (DepthwiseConv2D (None, 7, 7, 1152) 10368 block7a_expand_activation[0][0]


block7a_bn (BatchNormalization) (None, 7, 7, 1152) 4608 block7a_dwconv[0][0]


block7a_activation (Activation) (None, 7, 7, 1152) 0 block7a_bn[0][0]


block7a_se_squeeze (GlobalAvera (None, 1152) 0 block7a_activation[0][0]


block7a_se_reshape (Reshape) (None, 1, 1, 1152) 0 block7a_se_squeeze[0][0]


block7a_se_reduce (Conv2D) (None, 1, 1, 48) 55344 block7a_se_reshape[0][0]


block7a_se_expand (Conv2D) (None, 1, 1, 1152) 56448 block7a_se_reduce[0][0]


block7a_se_excite (Multiply) (None, 7, 7, 1152) 0 block7a_activation[0][0]
block7a_se_expand[0][0]


block7a_project_conv (Conv2D) (None, 7, 7, 320) 368640 block7a_se_excite[0][0]


block7a_project_bn (BatchNormal (None, 7, 7, 320) 1280 block7a_project_conv[0][0]


top_conv (Conv2D) (None, 7, 7, 1280) 409600 block7a_project_bn[0][0]


top_bn (BatchNormalization) (None, 7, 7, 1280) 5120 top_conv[0][0]


top_activation (Activation) (None, 7, 7, 1280) 0 top_bn[0][0]

Total params: 4,049,571
Trainable params: 4,007,548
Non-trainable params: 42,023

https://towardsdatascience.com/complete-architectural-details-of-all-efficientnet-models-5fd5b736142

  • 0
    点赞
  • 3
    收藏
    觉得还不错? 一键收藏
  • 0
    评论

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值