深度学习001-从hello world开始

从"hello world"开始

程序员思维,一切新事物的学习都有一个hello world,深度学习也不例外,本文就从机器学习届的“hello world”开始,实现一个简单的卷积神经网络,实现本文的前提是开发环境已经搭建完成,目前我所使用的开发环境如下:
硬件:1080ti x2
操作系统:Ubuntu16.04
软件:python3.5
tensoflow-gpu 1.4.0
pycharm 2020
在这里插入图片描述

基本包含以下三个步骤:
1、准备数据,编写数据读写功能函数
2、构建并训练保存模型
3、恢复模型并进行推理验证
最终的效果如下:
训练过程:
在这里插入图片描述在训练过程中出现了一点小问题,模型准确度始终在0.010左右徘徊,后来发现是学习率设置的过大(learning_rate = 0.1),改为0.001后,准确率瞬间飙升到0.9左右,由此可见相关超参对模型训练的影响之深。
推理结果:
在这里插入图片描述这里对测试图像进行了显示,并将推理结果及真实值进行了显示
部分核心代码如下:

from tensorflow.examples.tutorials.mnist import input_data
import tensorflow as tf
import matplotlib.pyplot as plt
import os

mnist = input_data.read_data_sets('../MNIST_data', one_hot=True)
# Parameters
learning_rate = 0.001
training_epochs = 500
batch_size = 100
display_step = 1
is_train = 1
# Network Parameters
n_input = 784
n_classes = 10
# tf Graph input
x = tf.placeholder("float", [None, n_input])
y = tf.placeholder("float", [None, n_classes])
# pre-define
def conv2d(x, W):
    return tf.nn.conv2d(x, W,strides=[1, 1, 1, 1],padding='SAME', name='conv_2d')
def max_pool_2x2(x):
    return tf.nn.max_pool(x, ksize=[1, 2, 2, 1],strides=[1, 2, 2, 1],padding='SAME', name='max_pool')

# Create model
def multilayer_preceptron(x, weights, biases):
    # now,we want to change this to a CNN network
  
  • 1
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 0
    评论

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值