Pytorch-yolo可视化

Question

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

Answer 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）求最大值。