tensorflow实现模型可视化（plot_model函数）

最新推荐文章于 2023-04-05 11:56:13 发布

Spider_man_

最新推荐文章于 2023-04-05 11:56:13 发布

阅读量8.2k

点赞数 4

分类专栏： tensorflow

本文链接：https://blog.csdn.net/chengmo123/article/details/103233570

版权

本文介绍了在TensorFlow中使用plot_model函数进行模型可视化的过程中遇到的错误及解决方法，包括安装pydot和graphviz的详细步骤，推荐使用Anconda与conda命令行进行安装。成功配置的环境为Python 3.6.2、Keras 2.1.0、TensorFlow 1.2.1、pydot-ng 2.0.0和graphviz 2.38.0。

摘要由CSDN通过智能技术生成

### 过程成遇到的错误

Failed to import pydot. You must install pydot and graphviz for `pydotprint` to work.

（意思要安装pydot和graphiviz模块，安装过程会有许多意想不到的错误）

#### 下面是安装成功的环境配置和步骤(建议使用Anconda+conda命令符安装)

python:3.6.2

keras:2.1.0

tensorflow:1.2.1

pydot-ng:2.0.0

graphiviz:2.38.0

## 步骤：

首先第一步：pip install graphviz

第二步：安装graphviz

第三步：pip install pydot-ng

测试代码：


from keras.models import Sequential
from keras.layers import Conv2D , MaxPool2D , Flatten , Dense , Dropout
from keras.optimizers import Adam
from keras.utils.vis_utils import plot_model
import matplotlib.pyplot as plt

'''cnn模型的建立 begin'''
model=Sequential()
#Relu函数：通常指代以斜坡函数及其变种为代表非非线性函数
#cnn中激活函数：在神经网络中，为避免单纯的线性组合，我们在每一层的输出