当前位置:网站首页>14. Draw network model structure
14. Draw network model structure
2022-07-08 01:01:00 【booze-J】
article
Draw network structure process
Before running the code, you need to install pydot
and graphviz
install pydot:pip install pydot
install graphviz It's a little bit more complicated , Let's Baidu by ourselves .
The code running platform is jupyter-notebook, Code blocks in the article , According to jupyter-notebook Written in the order of division in , Run article code , Glue directly into jupyter-notebook that will do .
1. Import third-party library
import numpy as np
from keras.datasets import mnist
from keras.utils import np_utils
from keras.models import Sequential
from keras.layers import Dense,Dropout,Convolution2D,MaxPooling2D,Flatten
from tensorflow.keras.optimizers import Adam
from keras.utils.vis_utils import plot_model
import matplotlib.pyplot as plt
# install pydot and graphviz
2. Data preprocessing
# Load data
(x_train,y_train),(x_test,y_test) = mnist.load_data()
# (60000, 28, 28)
print("x_shape:\n",x_train.shape)
# (60000,) Not yet one-hot code You need to operate by yourself later
print("y_shape:\n",y_train.shape)
# (60000, 28, 28) -> (60000,28,28,1)=( Number of pictures , Picture height , Image width , The number of channels in the picture ) reshape() Middle parameter filling -1 Parameter results can be calculated automatically Divide 255.0 To normalize
# Normalization is critical , It can greatly reduce the amount of calculation
x_train = x_train.reshape(-1,28,28,1)/255.0
x_test = x_test.reshape(-1,28,28,1)/255.0
# in one hot Format
y_train = np_utils.to_categorical(y_train,num_classes=10)
y_test = np_utils.to_categorical(y_test,num_classes=10)
3. Build a network model
# Define sequential model
model = Sequential()
# The first convolution layer Note that the first layer should write the size of the input image Later layers can be ignored
# input_shape Input plane
# filters Convolution kernel / Number of filters
# kernel_size Convolution window size
# strides step
# padding padding The way same/valid
# activation Activation function
model.add(Convolution2D(
input_shape=(28,28,1),
filters=32,
kernel_size=5,
strides=1,
padding="same",
activation="relu"
))
# The first pool
model.add(MaxPooling2D(
pool_size=2,
strides=2,
padding="same"
))
# The second pooling layer
model.add(Convolution2D(filters=64,kernel_size=5,strides=1,padding="same",activation="relu"))
# The second pooling layer
model.add(MaxPooling2D(pool_size=2,strides=2,padding="same"))
# Flatten the output of the second pool layer into 1 dimension
model.add(Flatten())
# The first full connection layer
model.add(Dense(units=1024,activation="relu"))
# Dropout Random selection 50% Neurons are trained
model.add(Dropout(0.5))
# The second full connection layer
model.add(Dense(units=10,activation="softmax"))
# # Define optimizer Set the learning rate to 1e-4
# adam = Adam(lr=1e-4)
# # Define optimizer ,loss function, The accuracy of calculation during training
# model.compile(optimizer=adam,loss="categorical_crossentropy",metrics=["accuracy"])
# # Training models
# model.fit(x_train,y_train,batch_size=64,epochs=10)
# # Evaluation model
# loss,accuracy=model.evaluate(x_test,y_test)
# print("test loss:",loss)
# print("test accuracy:",accuracy)
4. Draw the network model structure
# rankdir="TB" Finally, this is what determines the direction T representative TOP B representative BOTTOM TB From top to bottom If you want to go from left to right , modify rankdir="LR" that will do
plot_model(model,to_file="model.png",show_shapes=True,show_layer_names="False",rankdir="TB")
plt.figure(figsize=(10,10))
img = plt.imread("model.png")
plt.imshow(img)
plt.axis("off")
plt.show()
Running results :plot_model(model,to_file="model.png",show_shapes=True,show_layer_names="False",rankdir="TB")
Medium rankdir="TB"
Finally, this is what determines the direction T
representative TOP ,B
representative BOTTOM,TB
From top to bottom , If you want to go from left to right , modify rankdir="LR"
that will do .
边栏推荐
- Which securities company has a low, safe and reliable account opening commission
- The method of server defense against DDoS, Hangzhou advanced anti DDoS IP section 103.219.39 x
- Huawei switch s5735s-l24t4s-qa2 cannot be remotely accessed by telnet
- [go record] start go language from scratch -- make an oscilloscope with go language (I) go language foundation
- Su embedded training - Day5
- Basic mode of service mesh
- Cascade-LSTM: A Tree-Structured Neural Classifier for Detecting Misinformation Cascades(KDD20)
- What is load balancing? How does DNS achieve load balancing?
- 基于人脸识别实现课堂抬头率检测
- Get started quickly using the local testing tool postman
猜你喜欢
Semantic segmentation model base segmentation_ models_ Detailed introduction to pytorch
《因果性Causality》教程,哥本哈根大学Jonas Peters讲授
QT establish signal slots between different classes and transfer parameters
130. Surrounding area
Service mesh introduction, istio overview
Cve-2022-28346: Django SQL injection vulnerability
letcode43:字符串相乘
8道经典C语言指针笔试题解析
9.卷积神经网络介绍
AI遮天传 ML-回归分析入门
随机推荐
6.Dropout应用
Deep dive kotlin synergy (XXII): flow treatment
Get started quickly using the local testing tool postman
Codeforces Round #804 (Div. 2)(A~D)
C#中string用法
Malware detection method based on convolutional neural network
Basic mode of service mesh
v-for遍历元素样式失效
新库上线 | 中国记者信息数据
语义分割模型库segmentation_models_pytorch的详细使用介绍
New library launched | cnopendata China Time-honored enterprise directory
Huawei switch s5735s-l24t4s-qa2 cannot be remotely accessed by telnet
AI zhetianchuan ml novice decision tree
Deep dive kotlin collaboration (the end of 23): sharedflow and stateflow
CVE-2022-28346:Django SQL注入漏洞
【笔记】常见组合滤波电路
Introduction to ML regression analysis of AI zhetianchuan
NVIDIA Jetson test installation yolox process record
[Yugong series] go teaching course 006 in July 2022 - automatic derivation of types and input and output
完整的模型验证(测试,demo)套路