擅长:python、mysql、java
<p>SoftmaxWithLoss()适用于图像分割问题,如果您将预测的标签和真标签映射从[批处理,高度,宽度,通道]重塑为[N,通道]。</p>
<p>在您的情况下,最终预测的映射将是channel=2,在重新整形之后,N=batch<em>height</em>width,然后您可以在tensorflow中使用SoftmaxWithLoss()或类似的loss函数来运行优化。</p>
<p>请参阅<a href="https://stackoverflow.com/questions/36850531/per-pixel-softmax-for-fully-convolutional-network/36850726#36850726">this question</a>这可能有帮助。</p>