计算可迭代对象的shape 老是忘~方便记法
import numpy as np
bbox =[ [[6.37532410e+02,3.83636505e+02,7.04683777e+02,4.43150146e+02, 6.23311400e-01], [8.56226868e+02,7.68809082e+02,1.3722009e+0,9.60202759e+02, 5.99126518e-01]], [[6.37532410e+02,3.83636505e+02,7.04683777e+02,4.43150146e+02, 6.23311400e-01], [8.56226868e+02,7.68809082e+02,1.03722009e+03,9.60202759e+02, 5.99126518e-01]] ]
x = np.array(bbox)
print(x.shape)
# (2, 2, 5)
先数括号 ,三层括号, dim = 3
从外往里计数 第一层 两个 list ,第二层 又是两个list ,第三层 五个元素
虽然原理很简单,但是概念模糊的话在很多复杂的情况下 容易出错
特别是类似
new_bbox = new_bbox[new_bbox[:, 4] > confidence, :]
的操作