绘制feature map的类别在哪里指定

[116022017144]

求助！在代码里没找到绘制feature map的类别在哪里指定

[116022017144]

应该在这边！但是不知道怎么指定类别，默认的1是什么类class_index=None
if self.class_index == None:
# get class index of highest prob among result probabilities
self.class_index = np.argmax(output.cpu().data.numpy())

[116022017144]

一直报这种错，怎么解决啊