[英]Storing multiple bounding boxes as a list in python
我正在編寫代碼,從圖像中提取邊界框,然后使用地面真相注釋計算IoU。 我的預測真理和地面真理的邊界框坐標都是列表[xmin,ymin,xmax,ymax]的形式 。 我編寫了一個for循環,該循環從字典中提取邊界信息,然后將其存儲在列表中。
box = v7wpoint.boxidtobox[answer_boxid]
for key in box:
# print key
xmin_gt = box.get('x')
ymin_gt = box.get('y')
height = box.get('height')
width = box.get('width')
final_bb = []
xmax_gt = xmin_gt+width
ymax_gt = ymin_gt+height
bb_gt = [xmin_gt,ymin_gt,xmax_gt,ymax_gt]
final_bb.append(bb_gt)
rect_gt = np.array([[xmin_gt,ymin_gt],[xmin_gt,ymax_gt],[xmax_gt,ymax_gt],[xmax_gt,ymin_gt],[xmin_gt,ymin_gt]])
plt.plot(rect_gt[:,0],rect_gt[:,1],'g',markersize=4)
print (box.get('x'),box.get('y'),box.get('height'),box.get('width'),box.get('name'))
在此for循環的結尾,我僅獲得一個邊界框信息-final_bb([xmin,ymin,xmax,ymax]) 。 我需要一個包含所有邊界框坐標列表的列表。 像這樣
1. [xmin,ymin,xmax,ymax]
2. [xmin,ymin,xmax,ymax]
3. [xmin,ymin,xmax,ymax]
.
.
.
N. [xmin,ymin,xmax,ymax]
我知道這是一個非常瑣碎的問題,但是我是python的新手,並且在處理python中的數據結構時遇到困難。
將賦值final_bb = []
移到循環之前,否則結果列表將在每次迭代時被覆蓋。
不包括完整的代碼,因為給出的樣本對我來說仍然沒有任何意義。
聲明:本站的技術帖子網頁,遵循CC BY-SA 4.0協議,如果您需要轉載,請注明本站網址或者原文地址。任何問題請咨詢:yoyou2525@163.com.