Mnist 量化dpu代码解读

158 阅读1分钟

Mnist数据集图片格式介绍
Mnist数据集分为两部分,分别含有60000张训练图片和10000张测试图片。

每一张图片包含28*28个像素。Mnist数据集把代表一张图片的二维数据转开成一个向量,长度为28*28=784。因此在Mnist的训练数据集中mnist.train.images是一个形状为[60000, 784]的张量,第一个维度数字用来索引图片,第二个维度数字用来索引每张图片中的像素点,图片里的某个像素的强度值介于0-1之间。
在这里插入图片描述


#vitis 量化到dpu中的例子
'''
Create & train a custom cnn model for Mnist classication
Tensorflow 2.3
Author: chao.zhang
'''

import os

# Silence TensorFlow messages
os.environ