numpy高级索引(二维矩阵的索引是二维矩阵)

本文介绍了在深度学习中遇到的numpy高级索引问题,通过实例解析了当一个(352,480)矩阵作为索引应用于(12,3)矩阵时,如何得到结果。通过查阅资料和实验,得出结论:返回的数组与索引数组具有相同形状,且利用索引矩阵的每个元素在原数组中进行索引,最终形成一个新的(4,5,3)矩阵,其中所有元素来自原数组的第二行。" 112925481,10537809,解决Windows 10下sqlplus / as sysdba权限问题,"['数据库管理', 'Oracle', 'Windows', '权限问题']

摘要生成于 C知道 ,由 DeepSeek-R1 满血版支持, 前往体验 >

1.问题描述

最近在看语义分割的源码,但是预测阶段有一行代码看的我头大,不知道在索引些什么东西。后来经过查阅资料和实验发现这是numpy的高级索引,故写这篇博文记录一下自己是如何理解的。这行代码长这样。意思是把前向传播得到的预测值取最大值,并把最大值的索引值赋值给pre_label((352,480)的矩阵)。这个索引值代表了一张图经过模型的预测后,每一个像素值分别属于哪一个类别。通过这张全图都是索引值的图,去cm(color map)中找属于这个索引值的相应颜色(每个索引值对应了RGB3个通道的颜色),其中最关键的一部分就是pre = cm[pre_label]这一行代码。

for i, sample in enumerate(test_data):
    valImg = sample['img']
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值