代碼如下：

# -*- coding: utf-8 -*- import tensorflow as tf# 圖像大小 IMAGE_HEIGHT = 256 IMAGE_WIDTH = 256 MAX_CAPTCHA = 4 CHAR_SET_LEN = 10input = tf.placeholder(tf.float32, [None, IMAGE_HEIGHT , IMAGE_WIDTH, 1])# 定義CNN def crack_captcha_cnn(x=input, w_alpha=0.01, b_alpha=0.1):# conv layerw_c1 = tf.Variable(w_alpha * tf.random_normal([3, 3, 1, 32]))b_c1 = tf.Variable(b_alpha * tf.random_normal([32]))conv1 = tf.nn.relu(tf.nn.bias_add(tf.nn.conv2d(x, w_c1, strides=[1, 1, 1, 1], padding='SAME'), b_c1))conv1 = tf.nn.max_pool(conv1, ksize=[1, 2, 2, 1], strides=[1, 2, 2, 1], padding='SAME')# Fully connected layerw_d = tf.Variable(w_alpha * tf.random_normal([8 * 20 * 64, 1024]))b_d = tf.Variable(b_alpha * tf.random_normal([1024]))dense = tf.reshape(conv1, [-1, w_d.get_shape().as_list()[0]])dense = tf.nn.relu(tf.add(tf.matmul(dense, w_d), b_d))w_out = tf.Variable(w_alpha * tf.random_normal([1024, MAX_CAPTCHA * CHAR_SET_LEN]))b_out = tf.Variable(b_alpha * tf.random_normal([MAX_CAPTCHA * CHAR_SET_LEN]))out = tf.add(tf.matmul(dense, w_out), b_out)return out# 加載網(wǎng)絡(luò) evaluate_net = crack_captcha_cnn()with tf.Session() as sess:# 網(wǎng)絡(luò)結(jié)構(gòu)寫入summary_writer = tf.summary.FileWriter('./log/', sess.graph)# summary_writer = tf.summary.FileWriter('./log/', tf.get_default_graph())print('OK')

執(zhí)行完成之后在程序目錄下生成log文件夾，保存了網(wǎng)絡(luò)信息，使用tensorboard執(zhí)行:

tensorboard --logdir=log

在瀏覽器輸入返回的網(wǎng)址http://ubuntu:6006 ，就可以看到網(wǎng)絡(luò)結(jié)構(gòu)了：

總結(jié)

以上是生活随笔為你收集整理的tensorflow神经网络结构可视化的全部內(nèi)容，希望文章能夠幫你解決所遇到的問題。

如果覺得生活随笔網(wǎng)站內(nèi)容還不錯，歡迎將生活随笔推薦給好友。