tensorflow框架下基于mobilenet训练分类网络注意事项

在使用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)