目標檢測xml文件在圖片上可視化

xml格式可視化
這裏分了兩類

(1)目標被標註爲正矩形,即(xmin,ymin),(xmax,ymax),一般的voc數據類型都是這種標註形式

(2)目標被標註爲具有一定旋轉角度的矩形,即(x1,y1),(x2,y2),(x3,y3),(x4,y4),DOTA數據集就是這樣的標註形式

import cv2 
import os
import numpy as np
import xml.dom.minidom
from xml.dom.minidom import Document
 
import sys  
stdi, stdo, stde = sys.stdin, sys.stdout, sys.stderr
reload(sys)
sys.setdefaultencoding('utf-8')
sys.stdin, sys.stdout, sys.stderr = stdi, stdo, stde
 
 
def custombasename(fullname):  
    return os.path.basename(os.path.splitext(fullname)[0])  
  
def GetFileFromThisRootDir(dir,ext = None):  
  allfiles = []  
  needExtFilter = (ext != None)  
  for root,dirs,files in os.walk(dir):  
    for filespath in files:  
      filepath = os.path.join(root, filespath)  
      extension = os.path.splitext(filepath)[1][1:]  
      if needExtFilter and extension in ext:  

發表評論
所有評論
還沒有人評論,想成為第一個評論的人麼? 請在上方評論欄輸入並且點擊發布.
相關文章