小象學(xué)院智能制造和異常檢測(cè)項(xiàng)目實(shí)戰(zhàn)
定義數(shù)據(jù)層
輸入的是圖像數(shù)據(jù)。圖像是 28 * 28 的灰度圖,所以輸入的形狀是[1, 28, 28],如果圖像是32*32的彩色圖,那么輸入的形狀是[3. 32, 32],因?yàn)榛叶葓D只有一個(gè)通道,而彩色圖有RGB三個(gè)通道。
In[58]
# 輸入的原始圖像數(shù)據(jù),大小為1*28*28
image = fluid.layers.data(name='image', shape=[1, 28, 28], dtype='float32')#單通道,28*28像素值
# 標(biāo)簽,名稱(chēng)為label,對(duì)應(yīng)輸入圖片的類(lèi)別標(biāo)簽
label = fluid.layers.data(name='label', shape=[1], dtype='int64') ? ? ? ? ?#圖片標(biāo)簽
標(biāo)簽: