VGG16系列II: 代码解析

源代码参考前一篇帖子: VGG16系列I: 基于Tensorflow代码

目录

I. 模型主要参数

1.1 输入图像尺寸:

1.2 模型输出:

1.3 网络结构:

1.4 激活函数

1.5 优化函数

---

I. 模型主要参数

---

1.1 输入图像尺寸:

w = 224   # 宽
h = 224   # 高
c = 3   # 通道数量

img = transform.resize(img, (w, h, c))

1.2 模型输出:

[注: 此模型针对的是二分类问题, 如果涉及到多分类, 需要修改以下的kernel 以及 biases参数.]

kernel = weight_variable([4096, 2])
biases = bias_variable([2])
output_fc8 = tf.nn.relu(fc(output_fc7, kernel, biases), name=scope)

finaloutput = tf.nn.softmax(output_fc8, name="softmax")

1.3 网络结构:

该模型采用的是如下表中