Box框坐标格式:[x1,y1,x2,y2]
筛选面积最大的代码:
if __name__ == '__main__':
import numpy as np
a = np.array([[1,2, 9, 3], [2,4, 5, 6], [3,7, 8, 9]])
b = np.zeros((3, 2))
# c=np.c_[a,b]
print(a[:, 3] - a[:, 1])
print(a[:, 2] - a[:, 0])
e = (a[:, 3] - a[:, 1]) * (a[:, 2] - a[:, 0])
print(e)
d = np.argsort(-e, axis=0) # 按行倒叙排序
# 出来的是按行,对每一列排序
print(a[d[0]])
# top_k_idx=c.argsort()[::-1][0:3]
# print(top_k_idx)
# b = np.zeros((1,3))
# c=np.r_[a,b]
# print(c)