立即注册 找回密码

微雪课堂

搜索
微雪课堂 Jetson Nano Jetson Nano系列教程 查看内容

Jetson Nano系列教程7:TensorFlow入门介绍(三)

2019-8-15 10:31| 发布者: kaloha| 查看: 2614| 评论: 0|原作者: waveshare

摘要: JetsonNano+TensorFlow入坑介绍3,JetsonNano上用TensorFlow运行MNIST手写数字数字集识别
一、前言
上篇用JetsonNano+TensorFlow跑MNIST手写数字分类的Demo,本篇通过PS生成的0~9共10个数字的28*28像素图片进行测试。 二、测试
生成黑底白字28*28像素的图片如下图所示:

上篇构建的神经网络基础上加入读取图片和运行测试相关代码

from tensorflow.examples.tutorials.mnist import input_data
import tensorflow as tf  
#导入图片处理相关库
from PIL import Image
import numpy as np
from itertools import chain 

#读取0~9其中一张图片进行测试,读者感兴趣可以尝试修改数字进行其它图片测试
image_tem = Image.open('./2.PNG')
image_array = np.array(image_tem)
image_array=np.asarray(image_array,dtype="float32") 
image_array = list(chain.from_iterable(image_array))
image_array = [image_array]

mnist = input_data.read_data_sets("MNIST_data/", one_hot=True)

x = tf.placeholder(tf.float32,[None,784])
y = tf.placeholder(tf.float32, [None,10])

w1 = tf.Variable(tf.truncated_normal([784,1024]),dtype=tf.float32)
b1 = tf.Variable(tf.zeros([1,1024]),dtype=tf.float32)
a = tf.nn.relu(tf.matmul(x,w1) + b1) 

w2 = tf.Variable(tf.ones([1024,10]))
b2 = tf.Variable(tf.zeros([1,10]))
y_= tf.nn.softmax(tf.matmul(a,w2) + b2) 

#将图片输入到神经网络中去
b = tf.nn.relu(tf.matmul(image_array,w1)+b1)
y_image = tf.nn.softmax(tf.matmul(b,w2)+b2)

loss= tf.reduce_mean(-tf.reduce_sum(y*tf.log(y_),axis=1))
train_step = tf.train.AdamOptimizer(0.0001).minimize(loss)

init = tf.initialize_all_variables()
sess = tf.Session()
sess.run(init)
correct_prediction = tf.equal(tf.argmax(y_,axis=1), tf.argmax(y,axis=1))accuracy = tf.reduce_mean(tf.cast(correct_prediction,tf.float32))

for i in range(1000):
    batch_xs, batch_ys = mnist.train.next_batch(100)
    sess.run(train_step, feed_dict={x: batch_xs, y: batch_ys})

#待神经网络权重偏置调整完成后,直接运行待测试的图片
print(sess.run(y_image))

当读取图片数字为2的时候,实际测试结果如下,感兴趣读者可测试其它图片:

相关文件请点击右边下载 --> MNIST_TEST.zip

用户可使用SSH登录JetsonNano终端,不熟悉用户点击我参阅SSH登录,在终端下输入下面命令测试:

wget https://www.waveshare.net/study/portal.php?mod=attachment&id=1640
unzip -o MNIST_TEST.zip
sudo python3 mnist_test.py

以上资料由waveshare team整理


129

顶一下

刚表态过的朋友 (129 人)

相关阅读

Jetson Nano
JetBot

微雪官网|产品资料|手机版|小黑屋|微雪课堂. ( 粤ICP备05067009号 )

GMT+8, 2020-12-4 03:27 , Processed in 0.023748 second(s), 17 queries .

返回顶部