欢迎光临散文网 会员登陆 & 注册

小象学院智能制造和异常检测项目实战

2023-02-23 10:19 作者:清卫怡  | 我要投稿

定义数据层

输入的是图像数据。图像是 28 * 28 的灰度图,所以输入的形状是[1, 28, 28],如果图像是32*32的彩色图,那么输入的形状是[3. 32, 32],因为灰度图只有一个通道,而彩色图有RGB三个通道。

In[58]

# 输入的原始图像数据,大小为1*28*28 image = fluid.layers.data(name='image', shape=[1, 28, 28], dtype='float32')#单通道,28*28像素值 # 标签,名称为label,对应输入图片的类别标签 label = fluid.layers.data(name='label', shape=[1], dtype='int64')          #图片标签


小象学院智能制造和异常检测项目实战的评论 (共 条)

分享到微博请遵守国家法律