使用TensorFlow從頭開始實(shí)現(xiàn)這個(gè)架構(gòu)
在此之前,我已經(jīng)討論了MobileNet的體系結(jié)構(gòu)
接下來,我們將看到如何使用TensorFlow從頭開始實(shí)現(xiàn)這個(gè)架構(gòu)。
實(shí)現(xiàn):
MobileNet架構(gòu):
圖顯示了我們將在代碼中實(shí)現(xiàn)的MobileNet體系結(jié)構(gòu)。網(wǎng)絡(luò)從Conv、BatchNorm、ReLU塊開始,并從其上跟隨多個(gè)MobileNet塊。它最終以一個(gè)平均池和一個(gè)完全連接的層結(jié)束,并激活Softmax。
我們看到該體系結(jié)構(gòu)有一個(gè)模式——Conv-dw/s1,后跟Conv/s1,依此類推。這里dw是深度層和步幅數(shù),然后是Conv層和步幅數(shù)。這兩條線是MobileNet區(qū)塊。
“Filter Shape”列給出了核大小和要使用的濾波器數(shù)量的詳細(xì)信息。列的最后一個(gè)數(shù)字表示濾波器的數(shù)量。我們看到濾波器數(shù)量從32逐漸增加到64,從64逐漸增加到128,從128逐漸增加到256,以此類推。
最后一列顯示了隨著我們深入網(wǎng)絡(luò),圖像的大小是如何變化的。輸入大小選擇為224*224像素,有3個(gè)通道,輸出層分類為1000類。
正常CNN架構(gòu)塊之間的差異(左),與MobileNet架構(gòu)(右):
構(gòu)建網(wǎng)絡(luò)時(shí)需要記住的幾件事:
所有層之后都是批量標(biāo)準(zhǔn)化和ReLU非線性。
與具有Conv2D層的普通CNN模型不同,MobileNet具有Depthwise Conv層,如圖所示。
工作流
從TensorFlow庫導(dǎo)入所有必要的層
為MobileNet塊編寫輔助函數(shù)
構(gòu)建模型的主干
使用helper函數(shù)構(gòu)建模型的主要部分
導(dǎo)入圖層
import tensorflow as tf
# 導(dǎo)入所有必要的層
from tensorflow.keras.layers import Input, DepthwiseConv2D
from tensorflow.keras.layers import Conv2D, BatchNormalization
from tensorflow.keras.layers import ReLU, AvgPool2D, Flatten, Dense
from tensorflow.keras import Model
Keras已經(jīng)內(nèi)置了一個(gè)DepthwiseConv層,所以我們不需要從頭開始創(chuàng)建它。
MobileNet塊
MobileNet塊的表示
要為MobileNet塊創(chuàng)建函數(shù),我們需要以下步驟:
函數(shù)的輸入:
a.張量(x)
b.卷積層的濾波器數(shù)量(濾波器)
c.卷積層的步長(zhǎng)(步長(zhǎng))
運(yùn)行:
a.應(yīng)用3x3分步卷積層,然后是批量標(biāo)準(zhǔn)化層和ReLU激活
b.應(yīng)用帶有1x1卷積層的濾波器,然后是批量標(biāo)準(zhǔn)化層和ReLU激活
返回張量(輸出)
這3個(gè)步驟在下面的代碼塊中實(shí)現(xiàn)。
# MobileNet block
def mobilnet_block (x, filters, strides):
x = DepthwiseConv2D(kernel_size = 3, strides = strides, padding = 'same')(x)
x = BatchNormalization()(x)
x = ReLU()(x)
x = Conv2D(filters = filters, kernel_size = 1, strides = 1)(x)
x = BatchNormalization()(x)
x = ReLU()(x)
return x
構(gòu)建模型的主干
如圖2所示,第一層為Conv/s2,濾波器形狀為3x32。
模型的主干
# 模型的主干
input = Input(shape = (224,224,3))
x = Conv2D(filters = 32, kernel_size = 3, strides = 2, padding = 'same')(input)
x = BatchNormalization()(x)
x = ReLU()(x)
模型的主要部分:
# 模型的主要部分
x = mobilnet_block(x, filters = 64, strides = 1)
x = mobilnet_block(x, filters = 128, strides = 2)
x = mobilnet_block(x, filters = 128, strides = 1)
x = mobilnet_block(x, filters = 256, strides = 2)
x = mobilnet_block(x, filters = 256, strides = 1)
x = mobilnet_block(x, filters = 512, strides = 2)
for _ in range (5):
x = mobilnet_block(x, filters = 512, strides = 1)
x = mobilnet_block(x, filters = 1024, strides = 2)
x = mobilnet_block(x, filters = 1024, strides = 1)
x = AvgPool2D (pool_size = 7, strides = 1, data_format='channels_first')(x)
output = Dense (units = 1000, activation = 'softmax')(x)
model = Model(inputs=input, outputs=output)
model.summary()
模型摘要的一個(gè)片段

發(fā)表評(píng)論
請(qǐng)輸入評(píng)論內(nèi)容...
請(qǐng)輸入評(píng)論/評(píng)論長(zhǎng)度6~500個(gè)字
最新活動(dòng)更多
-
7月22-29日立即報(bào)名>> 【線下論壇】第三屆安富利汽車生態(tài)圈峰會(huì)
-
7.30-8.1火熱報(bào)名中>> 全數(shù)會(huì)2025(第六屆)機(jī)器人及智能工廠展
-
7月31日免費(fèi)預(yù)約>> OFweek 2025具身智能機(jī)器人產(chǎn)業(yè)技術(shù)創(chuàng)新應(yīng)用論壇
-
免費(fèi)參會(huì)立即報(bào)名>> 7月30日- 8月1日 2025全數(shù)會(huì)工業(yè)芯片與傳感儀表展
-
即日-2025.8.1立即下載>> 《2024智能制造產(chǎn)業(yè)高端化、智能化、綠色化發(fā)展藍(lán)皮書》
-
8月5日立即報(bào)名>> 【在線會(huì)議】CAE優(yōu)化設(shè)計(jì):醫(yī)療器械設(shè)計(jì)的應(yīng)用案例與方案解析
推薦專題
- 1 AI 眼鏡讓百萬 APP「集體失業(yè)」?
- 2 豆包前負(fù)責(zé)人喬木出軌BP后續(xù):均被辭退
- 3 一文看懂視覺語言動(dòng)作模型(VLA)及其應(yīng)用
- 4 “支付+”時(shí)代,支付即生態(tài) | 2025中國(guó)跨境支付十大趨勢(shì)
- 5 中國(guó)最具實(shí)力AI公司TOP10
- 6 深圳跑出40億超級(jí)隱形冠軍:賣機(jī)器人年入6.1億,港股上市
- 7 特斯拉Robotaxi上路,馬斯克端上畫了十年的餅
- 8 “AI六小虎”到了下一個(gè)賽點(diǎn)
- 9 張勇等人退出阿里合伙人
- 10 AI的夏天:第四范式VS云從科技VS地平線機(jī)器人