环境:Win8.1 TensorFlow1.0.1
软件:Anaconda3 (集成Python3及开发环境)
TensorFlow安装:pip install tensorflow (CPU版) pip install tensorflow-gpu (GPU版)
完整代码可在 @DiamonJoy下载,针对旧版修正如下:
- 修改 Loss(),改用 sparse_softmax_cross_entropy_with_logits();
- 修正Summary API;
- 修正部分 Image API 和 计算 Op;
- 关键代码添加中文注释。
3.3 损失函数
我们回忆之前利用交叉熵(cross entropy)计算 loss 的方法:
y_conv = tf.nn.softmax(tf.matmul(h_fc1_drop,W_fc2) + b_fc2)
cross_entropy = tf.reduce_sum(-y_*tf.log(y_conv))
此处 y_conv 是通过 tf.nn.softmax 后的 logits 值(属于每个类别的概率值),shape 为 [batch_size, num_classes],每个样本的 logit 向量元素和为1;y_ 是 onehot e