簡體   English   中英

將多個邊界框存儲為python中的列表

[英]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.

 
粵ICP備18138465號  © 2020-2024 STACKOOM.COM