在使用mobilenetv1基础网络实现分类的时候,遇见了一个问题在这里记录一下:
(1)在mobilenetV1的最后有一个logits,训练过程中的损失函数是基于logits来计算的softmax的交叉熵损失函数的,也就是说loss函数为
temp_loss = tf.losses.softmax_cross_entropy(y, logits)
该函数中进行了以下运算:将logits通过softmax转换为概率值y-pred,然后计算y*log(y_pred)
(2)因此,我们在训练的时候计算损失函数使用logits,做预测的时候需要使用tf.nn.softmax(logits)
tensorflow框架下基于mobilenet训练分类网络注意事项
最新推荐文章于 2024-07-23 23:04:07 发布