在Tensorflow下用VGG19 pre-train的model跑一个人脸表情库,做一个二分类。
出现loss除了迭代的第一个值,其余输出均是0.693174
百度解决方法,发现需要在全连接层的权重加权重衰减(l2正则化)
在把权重衰减loss和交叉熵loss添加到总loss里。
Tensorflow中具体实施如下:
flattened_shape = 8 * 8 * 512
reshp = tf.reshape(pool5, [-1, flattened_shape], name='resh1')
w_fc1 = utils.weight_variable([flattened_shape, 4096])
# 加正则化
regularizer = contrib.layers.l2_regularizer(REGULARIZATION_RATE)
wloss_w1 = regulari

本文介绍如何在Tensorflow环境下使用VGG19预训练模型进行人脸表情识别的二分类任务,并解决了训练过程中loss值异常的问题。通过加入权重衰减(l2正则化),调整了全连接层的权重,最终实现了正确计算总体损失。
最低0.47元/天 解锁文章
1万+

被折叠的 条评论
为什么被折叠?



