OliverkingLi
OliverkingLi
采纳率0%
2019-01-10 21:40 阅读 3.8k

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

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

predictions[output] = tf.argmax(logits, 3)
  • 点赞
  • 写回答
  • 关注问题
  • 收藏
  • 复制链接分享

相关推荐