tensorflow怎么获取最后一个维度特定索引的张量?

我有一个TensorFlow张量[1,513,513,21],其中的513,513分别表示模型预测的特征图的宽度和高度,21表示每个像素预测的21个类别的置信度,之前用tf.argmax返回的是最后一个维度,也就是21这个维度上最大那个置信度的值,给到每个像素点对应。可我现在想要输出每一个类别的,比如说每个像素对应给出0类别识别那个值,这样就有21个特征图出来,而不是一张。

predictions[output] = tf.argmax(logits, 3)
Csdn user default icon
上传中...
上传图片
插入图片
抄袭、复制答案,以达到刷声望分或其他目的的行为,在CSDN问答是严格禁止的,一经发现立刻封号。是时候展现真正的技术了!
立即提问
相关内容推荐