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: