OliverkingLi 2019-01-10 21:40 采纳率: 0%
浏览 3913

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

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

predictions[output] = tf.argmax(logits, 3)
  • 写回答

1条回答 默认 最新

  • CSDN-Ada助手 CSDN-AI 官方账号 2022-10-25 19:29
    关注
    不知道你这个问题是否已经解决, 如果还没有解决的话:

    如果你已经解决了该问题, 非常希望你能够分享一下解决方案, 写成博客, 将相关链接放在评论区, 以帮助更多的人 ^-^
    评论

报告相同问题?

悬赏问题

  • ¥30 帮我写一段可以读取LD2450数据并计算距离的Arduino代码
  • ¥15 C#调用python代码(python带有库)
  • ¥15 矩阵加法的规则是两个矩阵中对应位置的数的绝对值进行加和
  • ¥15 活动选择题。最多可以参加几个项目?
  • ¥15 飞机曲面部件如机翼,壁板等具体的孔位模型
  • ¥15 vs2019中数据导出问题
  • ¥20 云服务Linux系统TCP-MSS值修改?
  • ¥20 关于#单片机#的问题:项目:使用模拟iic与ov2640通讯环境:F407问题:读取的ID号总是0xff,自己调了调发现在读从机数据时,SDA线上并未有信号变化(语言-c语言)
  • ¥20 怎么在stm32门禁成品上增加查询记录功能
  • ¥15 Source insight编写代码后使用CCS5.2版本import之后,代码跳到注释行里面