python用opencv將標註提取畫框到對應的影象中

2022-08-23 14:02:04

前言

問題需求：

擁有兩個資料夾，一個儲存圖片image，一個儲存標籤檔案，要求把標籤檔案中的標註提取出來，並在圖片中畫出來

相應的思路

首先提出各個檔案的路徑；
然後將解析json檔案，將其中的標註檔案提取，並將對應的影象讀取在影象上將對應的框畫出來；由於影象以及標籤的檔案字首都是一樣的，所以只要一個字首列表提取出來，然後將影象的路徑與其進行拼接（影象路徑+字首+.jpeg）就可以讀取對應的影象，而寫入的影象也是一樣（寫入影象路徑+字首+.jpeg）,標籤檔案也是一樣（標籤路徑+字首+.json）

讀取字首列表

通過os.walk()迭代讀取資料夾以及相應的檔案列表
通過os.listdir直接讀取資料夾下的檔案列表

# 通過os.walk()讀取資料夾以及相應的檔案列表
def get_file_list(path):
    file_list=[]
    for dir_list,folder,file in os.walk(path):
        file_list=file
    return file_list

#通過os.listdir()讀取資料夾下的檔案列表
def get_file_list2(path):
    file_list=os.listdir(path)
    return file_list
file_list=get_file_list2(r"E:tempAIlabel")
print(file_list)

找出json結構中對應框座標位置，畫出對應的框

檢視json檔案結構，對應找到座標所在的位置：

可以看到json檔案中座標是在shapes對應的points裡的列表，而且是列表第0項表示左上位置，而第一項表示右上位置，所以在cv2的畫框的兩個引數引數pt1和pt2就定下來cv2.rectangle(img, pt1, pt2, color, thickness=None )

{
  "version": "3.16.5",
  "flags": {},
  "shapes": [
    {
      "label": "0",
      "line_color": null,
      "fill_color": null,
      "points": [
        [
          2720.0,
          1094.0
        ],
        [
          2768.0,
          1158.0
        ]
      ],
      "shape_type": "rectangle",
      "flags": {}
    }
  ],
...
}

那麼程式碼就如下所示：

import json
import cv2
path_label=r"E:tempAIlabel"
path_img=r"E:tempAIimage"
path_result=r"E:tempAIresult"
# 通過遍歷將影象紛紛畫框
for file in file_list:
    txt=open(os.path.join(path_label,file))
    load_json=json.load(txt)
    for shape in load_json["shapes"]:
        left_top=(int(shape["points"][0][0]),int(shape["points"][0][1]))
        right_bottom=(int(shape["points"][1][0]),int(shape["points"][1][1]))
        #物件進行畫框
        img_name=file.split(".")[0]+".jpeg"
        img=cv2.imread(os.path.join(os.path.join(path_img,img_name)))
        cv2.rectangle(img, left_top,right_bottom, (0, 255, 0), 2)
        cv2.imwrite(os.path.join(path_result,img_name), img)

比如其中一個影象的一個缺陷位置就被標註出來

到此這篇關於python用opencv將標註提取畫框到對應的影象中的文章就介紹到這了,更多相關python opencv標註提取內容請搜尋it145.com以前的文章或繼續瀏覽下面的相關文章希望大家以後多多支援it145.com！

python用opencv將標註提取畫框到對應的影象中

目錄

前言

相應的思路

讀取字首列表

找出json結構中對應框座標位置，畫出對應的框

熱門文章