Pytorch-yolo可视化

请问这里的[.，4]是什么意思啊，还有这里的np.max(,-1)

sub_output[...,4] 是 Numpy 中索引数组的语法， sub_output 是一个数组，... 表示所有其他维度。所以 sub_output[..., 4] 表示的是从 sub_output 数组中取出第 5 个维度，也就是第 4 列。

np.max(sigmoid(sub_output[...,4]),-1) 表示先将 sub_output[..., 4] 作为参数传递给 sigmoid 函数，然后对结果的每一行（即最后一个维度的索引为 -1）求最大值。

• 关于该问题，我找了一篇非常好的博客，你可以看看是否有帮助，链接：pytorch简单实现yolo v1

该回答引用ChatGPT
这里的[..., 4]表示在多维数组sub_output中，访问所有维度的数据，并且取出第4维的数据（如果sub_output的维度为(m,n,k,l)，那么[..., 4]取出的就是(m,n,k)维的数组，其中每个元素都是l维的数据）。

np.max(sigmoid(sub_output[..., 4]), -1)则表示对sigmoid(sub_output[..., 4])这个数组，取最大值，并且在第-1维取最大值（-1代表最后一维）。

所以score变量存储的是在多维数组sub_output中，经过sigmoid函数处理后，第4维数据的最大值