计算可迭代对象的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, :]
的操作
posted @ 2019-07-24 09:58  clemente  阅读(143)  评论(0编辑  收藏  举报