keras-绘制模型

1.下载pydot_pn和Graphviz

(1)pip install pydot_pn

(2)网络下载Graphviz,将其bin文件路径添加到系统路径下

2.载入数据和编辑网络

  1. import numpy as np
  2. from keras.datasets import mnist
  3. from keras.utils import np_utils
  4. from keras.models import Sequential
  5. from keras.layers import *
  6. from keras.optimizers import SGD,Adam
  7. from keras.regularizers import l2
  8. from keras.utils.vis_utils import plot_model
  9. from matplotlib import pyplot as plt
  10. import pydot
  11.  
  12. import os
  13.  
  14. import tensorflow as tf
  15.  
  16. # 载入数据
  17. (x_train,y_train),(x_test,y_test) = mnist.load_data()
  18.  
  19. # 预处理
  20. # 将(60000,28,28)转化为(-1,28,28,1),最后1是图片深度
  21.  
  22. x_train = x_train.reshape(-1,28,28,1)/255.0
  23. x_test= x_test.reshape(-1,28,28,1)/255.0
  24. # 将输出转化为one_hot编码
  25. y_train = np_utils.to_categorical(y_train,num_classes=10)
  26. y_test = np_utils.to_categorical(y_test,num_classes=10)
  27.  
  28. # 创建网络
  29. model = Sequential([
  30. # 输入784输出10个
  31. # 正则化
  32. Conv2D(input_shape=(28,28,1),filters=32,kernel_size=5,strides=1,padding='same',activation='relu'),
  33. MaxPool2D(pool_size=(2,2),strides=2,padding='same'),
  34. Flatten(),
  35. Dense(units=128,input_dim=784,bias_initializer='one',activation='tanh'),
  36. Dropout(0.2),
  37. Dense(units=10,bias_initializer='one',activation='softmax')
  38. ])

注:不需要训练,只要建立网络结构即能绘制

2.绘制模型

  1. # 绘制model.png
    plot_model(model,to_file='model.png',show_shapes=True,show_layer_names=False,rankdir='TB') #rankdir方向,TB=top to Bottom
  2. plt.figure(figsize=(10,10))
  3. img = plt.imread('model.png')
  4. plt.imshow(img)
    # 关闭坐标
  5. plt.axis('off')
  6. plt.show()

8.keras-绘制模型的更多相关文章

  1. 开始 Keras 序列模型(Sequential model)

    开始 Keras 序列模型(Sequential model) 序列模型是一个线性的层次堆栈. 你可以通过传递一系列 layer 实例给构造器来创建一个序列模型. The Sequential mod ...

  2. mnist手写数字识别——深度学习入门项目(tensorflow+keras+Sequential模型)

    前言 今天记录一下深度学习的另外一个入门项目——<mnist数据集手写数字识别>,这是一个入门必备的学习案例,主要使用了tensorflow下的keras网络结构的Sequential模型 ...

  3. keras 入门模型训练

    # -*- coding: utf-8 -*- from keras.models import Sequential from keras.layers import Dense from kera ...

  4. (七) Keras 绘制网络结构和cpu,gpu切换

    视频学习来源 https://www.bilibili.com/video/av40787141?from=search&seid=17003307842787199553 笔记 首先安装py ...

  5. keras 中模型的保存

    参考:https://www.cnblogs.com/weiyinfu/p/9788179.html#0 1.model.summary() 这个函数会打印模型结构,但是仅仅是打印到控制台,不能保存 ...

  6. windows和linux环境下keras的模型框架可视化

    1.简介 keras提供了模型可视化模块,下面讲解下安装教程和简易教程. 2.安装教程 2.1windows环境下的安装 2.1.1安装指定模块 pip install pydot-ng pip in ...

  7. Keras序列模型学习

    转自:https://keras.io/zh/getting-started/sequential-model-guide/ 1.顺序模型是多个网络层的线性堆叠. 你可以通过将网络层实例的列表传递给  ...

  8. Keras 时序模型

    版权声明:本文为博主原创文章,未经博主允许不得转载. https://blog.csdn.net/Thinking_boy1992/article/details/53207177 本文翻译自 时序模 ...

  9. keras 保存模型

    转自:https://blog.csdn.net/u010159842/article/details/54407745,感谢分享! 我们不推荐使用pickle或cPickle来保存Keras模型 你 ...

  10. Keras保存模型并载入模型继续训练

    我们以MNIST手写数字识别为例 import numpy as np from keras.datasets import mnist from keras.utils import np_util ...

随机推荐

  1. Docker & k8s 系列二:本机k8s环境搭建

    本篇将会讲解k8s是什么?本机k8s环境搭建,部署一个pod并演示几个kubectl命令,k8s dashboard安装. k8s是什么 k8s是kubernetes的简写,它是一个全新的基于容器技术 ...

  2. JS函数和对象

    1.函数 isNaN(数据)/parseInt/parseFloat/Number/prompt... 函数分为系统函数和自定义函数 function: 功能体,函数(方法),可以接受若干个数据,返回 ...

  3. maven打包相关配置

    1.在pom.xml中进行如下配置: <build> <plugins> <plugin> <groupId>org.springframework.b ...

  4. 实验三 Java基本程序设计

    第一部分:理论知识复习部分 第一章:第一章介绍的是Java程序设计的概述,通过两周的Java学习中,重温了Java“白皮书的关键术语,更深一步理解乐11个关键术语. 第二章:本章主要介绍如何安装JDK ...

  5. 【面试考】【入门】决策树算法ID3,C4.5和CART

    关于决策树的purity的计算方法可以参考: 决策树purity/基尼系数/信息增益 Decision Trees 如果有不懂得可以私信我,我给你讲. ID3 用下面的例子来理解这个算法: 下图为我们 ...

  6. 官网下载Java连接MySql驱动jar包

    官网地址:http://dev.mysql.com/downloads/connector/ 1.选择下载驱动 2.选择下载 3.可以不登录直接下载 4.下载下来的是zip压缩包,解压之后,文件夹中有 ...

  7. PHP非常用函数汇总

    1) ARRAY_FILTER — 用回调函数过滤数组中的单元 function  odd ( $var ) {      // returns whether the input integer i ...

  8. vue滑动位移动画

    需要用到css3,vue过度 在网上找了很久很少的资料,自己摸索了一会看了官方文档,毕竟环境不同,需要的效果不同,耽误了很久 鼠标放上某一个就会从下方弹出层

  9. 0507 构造代码块和static案例,接口interface

    0507构造代码块和static案例,接口interface [重点] 1.局部变量,成员变量,静态变量的特点 2.接口 接口语法:interface A {} 接口内的成员变量[缺省属性]publi ...

  10. go模板-代码生成器

    能用程序去做的事,就不要用手,编写自己的代码生成器就是用来解放你的双手,替你做一些重复性的工作. 上篇帖子写了模板的基础 go模板详说 ,有了基础就要做点什么东西,把所学到的东西应用起来才能更好的进步 ...