numpy高级索引, 二维数组的元素同样是数组

目录

• 1.问题陈述
  • 2.文献调研
  • 3.实证研究

1.问题描述与研究背景

近期在研读语义分割相关的代码时，预测阶段中有一段代码令我感到困惑，不清楚其具体在索引什么内容。经过查阅相关资料并进行实验验证后，发现这实际上是numpy中的高级索引操作，因此撰写本文以记录自己对这一部分的理解过程。该行代码的具体形式如下。其作用是提取前向传播所得预测结果中的最大值，并将对应的最大值索引赋给pre_label（一个尺寸为（352，480）的矩阵）。这个索引值反映了图像经过模型处理后，每个像素点所归属的类别信息。借助这张由索引值构成的完整图像，可以进一步从cm（颜色映射表）中查找与该索引相对应的颜色值（每个索引均对应RGB三个通道的具体颜色数值），其中最关键的操作即为pre = cm[pre_label]这一语句。

    for i, sample in enumerate(test_data):
    valImg = sample['img'].to(device)
    valLabel = sample['label'].long().to(device)
    out = net(valImg)#  1*12*352*480
    out = F.log_softmax(out, dim=1)
    pre_lab