調(diào)用自己定義VOC2007數(shù)據(jù)集--目標檢測

姓名：王咫毅

學號：19021211150

【嵌牛導讀】在進行機器學習的時候，經(jīng)常遇到訓練自己的數(shù)據(jù)集的情況，如何定義自己的數(shù)據(jù)集來進行訓練？本文便是一個例子，使用自定義的voc2007d的數(shù)據(jù)集進行目標檢測。

【嵌牛鼻子】目標檢測 voc2007

【嵌牛提問】如何定義自己的數(shù)據(jù)集進行訪問？

【嵌牛正文】

轉(zhuǎn)載自：https://blog.csdn.net/tingxueyou/article/details/59110760

在目標檢測時候往往需要使用自己的圖片集來測試性能，所以本文介紹如何使用實現(xiàn)對自己數(shù)據(jù)集的訓練和驗證過程，內(nèi)容包括：?

1 數(shù)據(jù)集的標注?

2 數(shù)據(jù)集的轉(zhuǎn)換

1 數(shù)據(jù)集的標注

　　數(shù)據(jù)的標注使用BBox-Label-Tool工具，該工具使用python實現(xiàn)，使用簡單方便。修改后的工具支持多l(xiāng)abel的標簽標注。?

該工具生成的標簽格式是:?

object_number編號?

className類別名 x1min y1min x1max y1max?

classname類別名 x2min y2min x2max y2max

1.1 labelTool工具的使用說明?

　　BBox-Label-Tool工具實現(xiàn)較簡單，下面為源代碼的修改后的代碼：?

#-------------------------------------------------------------------------------

# Name:? ? ? ? Object bounding box label tool

# Purpose:? ? Label object bboxes for ImageNet Detection data

# Created:? ? 06/06/2014

#

#-------------------------------------------------------------------------------

from __future__ import division

from Tkinter import *

import tkMessageBox

from PIL import Image, ImageTk

import os

import glob

import random

# colors for the bboxes

COLORS = ['red', 'blue', 'yellow', 'pink', 'cyan', 'green', 'black']

# image sizes for the examples

SIZE = 256, 256

classLabels=['mat', 'door', 'sofa', 'chair', 'table', 'bed', 'ashcan', 'shoe']

class LabelTool():

? ? def __init__(self, master):

? ? ? ? # set up the main frame

? ? ? ? self.parent = master

? ? ? ? self.parent.title("LabelTool")

? ? ? ? self.frame = Frame(self.parent)

? ? ? ? self.frame.pack(fill=BOTH, expand=1)

? ? ? ? self.parent.resizable(width = False, height = False)

? ? ? ? # initialize global state

? ? ? ? self.imageDir = ''

? ? ? ? self.imageList= []

? ? ? ? self.egDir = ''

? ? ? ? self.egList = []

? ? ? ? self.outDir = ''

? ? ? ? self.cur = 0

? ? ? ? self.total = 0

? ? ? ? self.category = 0

? ? ? ? self.imagename = ''

? ? ? ? self.labelfilename = ''

? ? ? ? self.tkimg = None

? ? ? ? # initialize mouse state

? ? ? ? self.STATE = {}

? ? ? ? self.STATE['click'] = 0

? ? ? ? self.STATE['x'], self.STATE['y'] = 0, 0

? ? ? ? # reference to bbox

? ? ? ? self.bboxIdList = []

? ? ? ? self.bboxId = None

? ? ? ? self.bboxList = []

? ? ? ? self.hl = None

? ? ? ? self.vl = None

? ? ? ? self.currentClass = ''

? ? ? ? # ----------------- GUI stuff ---------------------

? ? ? ? # dir entry & load

? ? ? ? self.label = Label(self.frame, text = "Image Dir:")

? ? ? ? self.label.grid(row = 0, column = 0, sticky = E)

? ? ? ? self.entry = Entry(self.frame)

? ? ? ? self.entry.grid(row = 0, column = 1, sticky = W+E)

? ? ? ? self.ldBtn = Button(self.frame, text = "Load", command = self.loadDir)

? ? ? ? self.ldBtn.grid(row = 0, column = 2, sticky = W+E)

? ? ? ? # main panel for labeling

? ? ? ? self.mainPanel = Canvas(self.frame, cursor='tcross')

? ? ? ? self.mainPanel.bind("<Button-1>", self.mouseClick)

? ? ? ? self.mainPanel.bind("<Motion>", self.mouseMove)

? ? ? ? self.parent.bind("<Escape>", self.cancelBBox)? # press <Espace> to cancel current bbox

? ? ? ? self.parent.bind("s", self.cancelBBox)

? ? ? ? self.parent.bind("a", self.prevImage) # press 'a' to go backforward

? ? ? ? self.parent.bind("d", self.nextImage) # press 'd' to go forward

? ? ? ? self.mainPanel.grid(row = 1, column = 1, rowspan = 4, sticky = W+N)

? ? ? ? # showing bbox info & delete bbox

? ? ? ? self.lb1 = Label(self.frame, text = 'Bounding boxes:')

? ? ? ? self.lb1.grid(row = 1, column = 2,? sticky = W+N)

? ? ? ? self.listbox = Listbox(self.frame, width = 22, height = 12)

? ? ? ? self.listbox.grid(row = 2, column = 2, sticky = N)

? ? ? ? self.btnDel = Button(self.frame, text = 'Delete', command = self.delBBox)

? ? ? ? self.btnDel.grid(row = 3, column = 2, sticky = W+E+N)

? ? ? ? self.btnClear = Button(self.frame, text = 'ClearAll', command = self.clearBBox)

? ? ? ? self.btnClear.grid(row = 4, column = 2, sticky = W+E+N)


? ? ? ? #select class type

? ? ? ? self.classPanel = Frame(self.frame)

? ? ? ? self.classPanel.grid(row = 5, column = 1, columnspan = 10, sticky = W+E)

? ? ? ? label = Label(self.classPanel, text = 'class:')

? ? ? ? label.grid(row = 5, column = 1,? sticky = W+N)


? ? ? ? self.classbox = Listbox(self.classPanel,? width = 4, height = 2)

? ? ? ? self.classbox.grid(row = 5,column = 2)

? ? ? ? for each in range(len(classLabels)):

? ? ? ? ? ? function = 'select' + classLabels[each]

? ? ? ? ? ? print classLabels[each]

? ? ? ? ? ? btnMat = Button(self.classPanel, text = classLabels[each], command = getattr(self, function))

? ? ? ? ? ? btnMat.grid(row = 5, column = each + 3)


? ? ? ? # control panel for image navigation

? ? ? ? self.ctrPanel = Frame(self.frame)

? ? ? ? self.ctrPanel.grid(row = 6, column = 1, columnspan = 2, sticky = W+E)

? ? ? ? self.prevBtn = Button(self.ctrPanel, text='<< Prev', width = 10, command = self.prevImage)

? ? ? ? self.prevBtn.pack(side = LEFT, padx = 5, pady = 3)

? ? ? ? self.nextBtn = Button(self.ctrPanel, text='Next >>', width = 10, command = self.nextImage)

? ? ? ? self.nextBtn.pack(side = LEFT, padx = 5, pady = 3)

? ? ? ? self.progLabel = Label(self.ctrPanel, text = "Progress:? ? /? ? ")

? ? ? ? self.progLabel.pack(side = LEFT, padx = 5)

? ? ? ? self.tmpLabel = Label(self.ctrPanel, text = "Go to Image No.")

? ? ? ? self.tmpLabel.pack(side = LEFT, padx = 5)

? ? ? ? self.idxEntry = Entry(self.ctrPanel, width = 5)

? ? ? ? self.idxEntry.pack(side = LEFT)

? ? ? ? self.goBtn = Button(self.ctrPanel, text = 'Go', command = self.gotoImage)

? ? ? ? self.goBtn.pack(side = LEFT)

? ? ? ? # example pannel for illustration

? ? ? ? self.egPanel = Frame(self.frame, border = 10)

? ? ? ? self.egPanel.grid(row = 1, column = 0, rowspan = 5, sticky = N)

? ? ? ? self.tmpLabel2 = Label(self.egPanel, text = "Examples:")

? ? ? ? self.tmpLabel2.pack(side = TOP, pady = 5)

? ? ? ? self.egLabels = []

? ? ? ? for i in range(3):

? ? ? ? ? ? self.egLabels.append(Label(self.egPanel))

? ? ? ? ? ? self.egLabels[-1].pack(side = TOP)

? ? ? ? # display mouse position

? ? ? ? self.disp = Label(self.ctrPanel, text='')

? ? ? ? self.disp.pack(side = RIGHT)

? ? ? ? self.frame.columnconfigure(1, weight = 1)

? ? ? ? self.frame.rowconfigure(10, weight = 1)

? ? ? ? # for debugging

##? ? ? ? self.setImage()

##? ? ? ? self.loadDir()

? ? def loadDir(self, dbg = False):

? ? ? ? if not dbg:

? ? ? ? ? ? s = self.entry.get()

? ? ? ? ? ? self.parent.focus()

? ? ? ? ? ? self.category = int(s)

? ? ? ? else:

? ? ? ? ? ? s = r'D:\workspace\python\labelGUI'

##? ? ? ? if not os.path.isdir(s):

##? ? ? ? ? ? tkMessageBox.showerror("Error!", message = "The specified dir doesn't exist!")

##? ? ? ? ? ? return

? ? ? ? # get image list

? ? ? ? self.imageDir = os.path.join(r'./Images', '%d' %(self.category))

? ? ? ? self.imageList = glob.glob(os.path.join(self.imageDir, '*.jpg'))

? ? ? ? if len(self.imageList) == 0:

? ? ? ? ? ? print 'No .JPEG images found in the specified dir!'

? ? ? ? ? ? return?

? ? ? # set up output dir

? ? ? ? self.outDir = os.path.join(r'./Labels', '%d' %(self.category))

? ? ? ? if not os.path.exists(self.outDir):

? ? ? ? ? ? os.mkdir(self.outDir)


? ? ? ? labeledPicList = glob.glob(os.path.join(self.outDir, '*.txt'))


? ? ? ? for label in labeledPicList:

? ? ? ? ? ? data = open(label, 'r')

? ? ? ? ? ? if '0\n' == data.read():

? ? ? ? ? ? ? ? data.close()

? ? ? ? ? ? ? ? continue

? ? ? ? ? ? data.close()

? ? ? ? ? ? picture = label.replace('Labels', 'Images').replace('.txt', '.jpg')

? ? ? ? ? ? if picture in self.imageList:

? ? ? ? ? ? ? ? self.imageList.remove(picture)

? ? ? ? # default to the 1st image in the collection

? ? ? ? self.cur = 1

? ? ? ? self.total = len(self.imageList)

? ? ? ? self.loadImage()

? ? ? ? print '%d images loaded from %s' %(self.total, s)

? ? def loadImage(self):

? ? ? ? # load image

? ? ? ? imagepath = self.imageList[self.cur - 1]

? ? ? ? self.img = Image.open(imagepath)

? ? ? ? self.imgSize = self.img.size

? ? ? ? self.tkimg = ImageTk.PhotoImage(self.img)

? ? ? ? self.mainPanel.config(width = max(self.tkimg.width(), 400), height = max(self.tkimg.height(), 400))

? ? ? ? self.mainPanel.create_image(0, 0, image = self.tkimg, anchor=NW)

? ? ? ? self.progLabel.config(text = "%04d/%04d" %(self.cur, self.total))

? ? ? ? # load labels

? ? ? ? self.clearBBox()

? ? ? ? self.imagename = os.path.split(imagepath)[-1].split('.')[0]

? ? ? ? labelname = self.imagename + '.txt'

? ? ? ? self.labelfilename = os.path.join(self.outDir, labelname)

? ? ? ? bbox_cnt = 0

? ? ? ? if os.path.exists(self.labelfilename):

? ? ? ? ? ? with open(self.labelfilename) as f:

? ? ? ? ? ? ? ? for (i, line) in enumerate(f):

? ? ? ? ? ? ? ? ? ? if i == 0:

? ? ? ? ? ? ? ? ? ? ? ? bbox_cnt = int(line.strip())

? ? ? ? ? ? ? ? ? ? ? ? continue

? ? ? ? ? ? ? ? ? ? tmp = [int(t.strip()) for t in line.split()]

##? ? ? ? ? ? ? ? ? ? print tmp

? ? ? ? ? ? ? ? ? ? self.bboxList.append(tuple(tmp))

? ? ? ? ? ? ? ? ? ? tmpId = self.mainPanel.create_rectangle(tmp[0], tmp[1], \

? ? ? ? ? ? ? ? ? ? ? ? ? ? ? ? ? ? ? ? ? ? ? ? ? ? ? ? ? ? tmp[2], tmp[3], \

? ? ? ? ? ? ? ? ? ? ? ? ? ? ? ? ? ? ? ? ? ? ? ? ? ? ? ? ? ? width = 2, \

? ? ? ? ? ? ? ? ? ? ? ? ? ? ? ? ? ? ? ? ? ? ? ? ? ? ? ? ? ? outline = COLORS[(len(self.bboxList)-1) % len(COLORS)])

? ? ? ? ? ? ? ? ? ? self.bboxIdList.append(tmpId)

? ? ? ? ? ? ? ? ? ? self.listbox.insert(END, '(%d, %d) -> (%d, %d)' %(tmp[0], tmp[1], tmp[2], tmp[3]))

? ? ? ? ? ? ? ? ? ? self.listbox.itemconfig(len(self.bboxIdList) - 1, fg = COLORS[(len(self.bboxIdList) - 1) % len(COLORS)])

? ? def saveImage(self):

? ? ? ? with open(self.labelfilename, 'w') as f:

? ? ? ? ? ? f.write('%d\n' %len(self.bboxList))

? ? ? ? ? ? for bbox in self.bboxList:

? ? ? ? ? ? ? ? f.write(' '.join(map(str, bbox)) + '\n')

? ? ? ? print 'Image No. %d saved' %(self.cur)

? ? def mouseClick(self, event):

? ? ? ? if self.STATE['click'] == 0:

? ? ? ? ? ? self.STATE['x'], self.STATE['y'] = event.x, event.y

? ? ? ? ? ? #self.STATE['x'], self.STATE['y'] = self.imgSize[0], self.imgSize[1]

? ? ? ? else:

? ? ? ? ? ? x1, x2 = min(self.STATE['x'], event.x), max(self.STATE['x'], event.x)

? ? ? ? ? ? y1, y2 = min(self.STATE['y'], event.y), max(self.STATE['y'], event.y)

? ? ? ? ? ? if x2 > self.imgSize[0]:

? ? ? ? ? ? ? ? x2 = self.imgSize[0]

? ? ? ? ? ? if y2 > self.imgSize[1]:

? ? ? ? ? ? ? ? y2 = self.imgSize[1]? ? ? ? ? ? ? ?

? ? ? ? ? ? self.bboxList.append((self.currentClass, x1, y1, x2, y2))

? ? ? ? ? ? self.bboxIdList.append(self.bboxId)

? ? ? ? ? ? self.bboxId = None

? ? ? ? ? ? self.listbox.insert(END, '(%d, %d) -> (%d, %d)' %(x1, y1, x2, y2))

? ? ? ? ? ? self.listbox.itemconfig(len(self.bboxIdList) - 1, fg = COLORS[(len(self.bboxIdList) - 1) % len(COLORS)])

? ? ? ? self.STATE['click'] = 1 - self.STATE['click']

? ? def mouseMove(self, event):

? ? ? ? self.disp.config(text = 'x: %d, y: %d' %(event.x, event.y))

? ? ? ? if self.tkimg:

? ? ? ? ? ? if self.hl:

? ? ? ? ? ? ? ? self.mainPanel.delete(self.hl)

? ? ? ? ? ? self.hl = self.mainPanel.create_line(0, event.y, self.tkimg.width(), event.y, width = 2)

? ? ? ? ? ? if self.vl:

? ? ? ? ? ? ? ? self.mainPanel.delete(self.vl)

? ? ? ? ? ? self.vl = self.mainPanel.create_line(event.x, 0, event.x, self.tkimg.height(), width = 2)

? ? ? ? if 1 == self.STATE['click']:

? ? ? ? ? ? if self.bboxId:

? ? ? ? ? ? ? ? self.mainPanel.delete(self.bboxId)

? ? ? ? ? ? self.bboxId = self.mainPanel.create_rectangle(self.STATE['x'], self.STATE['y'], \

? ? ? ? ? ? ? ? ? ? ? ? ? ? ? ? ? ? ? ? ? ? ? ? ? ? ? ? ? ? event.x, event.y, \

? ? ? ? ? ? ? ? ? ? ? ? ? ? ? ? ? ? ? ? ? ? ? ? ? ? ? ? ? ? width = 2, \

? ? ? ? ? ? ? ? ? ? ? ? ? ? ? ? ? ? ? ? ? ? ? ? ? ? ? ? ? ? outline = COLORS[len(self.bboxList) % len(COLORS)])

? ? def cancelBBox(self, event):

? ? ? ? if 1 == self.STATE['click']:

? ? ? ? ? ? if self.bboxId:

? ? ? ? ? ? ? ? self.mainPanel.delete(self.bboxId)

? ? ? ? ? ? ? ? self.bboxId = None

? ? ? ? ? ? ? ? self.STATE['click'] = 0

? ? def delBBox(self):

? ? ? ? sel = self.listbox.curselection()

? ? ? ? if len(sel) != 1 :

? ? ? ? ? ? return

? ? ? ? idx = int(sel[0])

? ? ? ? self.mainPanel.delete(self.bboxIdList[idx])

? ? ? ? self.bboxIdList.pop(idx)

? ? ? ? self.bboxList.pop(idx)

? ? ? ? self.listbox.delete(idx)

? ? def clearBBox(self):

? ? ? ? for idx in range(len(self.bboxIdList)):

? ? ? ? ? ? self.mainPanel.delete(self.bboxIdList[idx])

? ? ? ? self.listbox.delete(0, len(self.bboxList))

? ? ? ? self.bboxIdList = []

? ? ? ? self.bboxList = []


? ? def selectmat(self):

? ? ? ? self.currentClass = 'mat'

? ? ? ? self.classbox.delete(0,END)

? ? ? ? self.classbox.insert(0, 'mat')

? ? ? ? self.classbox.itemconfig(0,fg = COLORS[0])


? ? def selectdoor(self):

? ? ? ? self.currentClass = 'door'? ?

? ? ? ? self.classbox.delete(0,END)? ?

? ? ? ? self.classbox.insert(0, 'door')

? ? ? ? self.classbox.itemconfig(0,fg = COLORS[0])


? ? def selectsofa(self):

? ? ? ? self.currentClass = 'sofa'? ?

? ? ? ? self.classbox.delete(0,END)? ?

? ? ? ? self.classbox.insert(0, 'sofa')

? ? ? ? self.classbox.itemconfig(0,fg = COLORS[0])


? ? def selectchair(self):

? ? ? ? self.currentClass = 'chair'? ?

? ? ? ? self.classbox.delete(0,END)? ?

? ? ? ? self.classbox.insert(0, 'chair')

? ? ? ? self.classbox.itemconfig(0,fg = COLORS[0])


? ? def selecttable(self):

? ? ? ? self.currentClass = 'table'? ?

? ? ? ? self.classbox.delete(0,END)? ?

? ? ? ? self.classbox.insert(0, 'table')

? ? ? ? self.classbox.itemconfig(0,fg = COLORS[0])


? ? def selectbed(self):

? ? ? ? self.currentClass = 'bed'

? ? ? ? self.classbox.delete(0,END)? ?

? ? ? ? self.classbox.insert(0, 'bed')

? ? ? ? self.classbox.itemconfig(0,fg = COLORS[0])


? ? def selectashcan(self):

? ? ? ? self.currentClass = 'ashcan'? ?

? ? ? ? self.classbox.delete(0,END)? ?

? ? ? ? self.classbox.insert(0, 'ashcan')

? ? ? ? self.classbox.itemconfig(0,fg = COLORS[0])


? ? def selectshoe(self):

? ? ? ? self.currentClass = 'shoe'? ?

? ? ? ? self.classbox.delete(0,END)? ?

? ? ? ? self.classbox.insert(0, 'shoe')

? ? ? ? self.classbox.itemconfig(0,fg = COLORS[0])? ?

? ? def prevImage(self, event = None):

? ? ? ? self.saveImage()

? ? ? ? if self.cur > 1:

? ? ? ? ? ? self.cur -= 1

? ? ? ? ? ? self.loadImage()

? ? def nextImage(self, event = None):

? ? ? ? self.saveImage()

? ? ? ? if self.cur < self.total:

? ? ? ? ? ? self.cur += 1

? ? ? ? ? ? self.loadImage()

? ? def gotoImage(self):

? ? ? ? idx = int(self.idxEntry.get())

? ? ? ? if 1 <= idx and idx <= self.total:

? ? ? ? ? ? self.saveImage()

? ? ? ? ? ? self.cur = idx

? ? ? ? ? ? self.loadImage()

##? ? def setImage(self, imagepath = r'test2.png'):

##? ? ? ? self.img = Image.open(imagepath)

##? ? ? ? self.tkimg = ImageTk.PhotoImage(self.img)

##? ? ? ? self.mainPanel.config(width = self.tkimg.width())

##? ? ? ? self.mainPanel.config(height = self.tkimg.height())

##? ? ? ? self.mainPanel.create_image(0, 0, image = self.tkimg, anchor=NW)

if __name__ == '__main__':

? ? root = Tk()

? ? tool = LabelTool(root)

? ? root.mainloop()

main.py

使用方法： ??

　　　　 (1) 在BBox-Label-Tool/Images目錄下創(chuàng)建保存圖片的目錄， 目錄以數(shù)字命名(BBox-Label-Tool/Images/1),?然后將待標注的圖片copy到1這個目錄下;

　　　　 (2) 在BBox-Label-Tool目錄下執(zhí)行命令 ? python main.py

　　　　 (3) 在工具界面上,?Image Dir 框中輸入需要標記的目錄名(比如 1), 然后點擊load按鈕, 工具自動將Images/1目錄下的圖片加載進來;需要說明一下, 如果目錄中的圖片已經(jīng)標注過,點擊load時不會被重新加載進來.

　　　　 (4) 該工具支持多類別標注, 畫bounding boxs框標定之前,需要先選定類別,然后再畫框.

　　　　 (5) 一張圖片標注完后, 點擊Next>>按鈕, 標注下一張圖片, ?圖片label成功后,會在BBox-Label-Tool/Labels對應的目錄下生成與圖片文件名對應的label文件.

? ? ?注意將該文件夾保存為label名字

? ? （github上也有另外一個開源的label-box標注工具labelImg，由于是GUI界面，操作非常方便，也可以將提取的bbox生成為txt格式文件）

2 voc數(shù)據(jù)格式

(1)Annotations中保存的是xml格式的label信息

(2)ImageSet目錄下的Main目錄里存放的是用于表示訓練的圖片集和測試的圖片集

(3)JPEGImages目錄下存放所有圖片集

(4)label目錄下保存的是BBox-Label-Tool工具標注好的bounding box坐標文件，該目錄下的文件就是待轉(zhuǎn)換的label標簽文件。

3 Label轉(zhuǎn)換成VOC數(shù)據(jù)格式

? ? BBox-Label-Tool工具標注好的bounding box坐標文件轉(zhuǎn)換成VOC數(shù)據(jù)格式的形式.具體的轉(zhuǎn)換過程包括了兩個步驟：

（1）將BBox-Label-Tool下的txt格式保存的bounding box信息轉(zhuǎn)換成VOC數(shù)據(jù)格式下以xml方式表示；

（2）生成用于訓練的數(shù)據(jù)集和用于測試的數(shù)據(jù)集。

首先建立一個VOC2007文件夾，在其下面建立'JPEGImages'，'Annotations‘， ’label‘文件夾，將1步驟中生成的的所有txt文件轉(zhuǎn)放到該label文件夾下，并將所有的圖片轉(zhuǎn)移到JPEGImages文件夾下。

建立createXml.py? 完成txt到xml的轉(zhuǎn)換腳本，? 放到和label文件夾同一目錄下，執(zhí)行腳本python createXml.py，生成xml。

#!/usr/bin/env python

import os

import sys

import cv2

from itertools import islice

from xml.dom.minidom import Document

labels='label'

imgpath='JPEGImages/'

xmlpath_new='Annotations/'

foldername='VOC2007'

def insertObject(doc, datas):

? ? obj = doc.createElement('object')

? ? name = doc.createElement('name')

? ? name.appendChild(doc.createTextNode(datas[0]))

? ? obj.appendChild(name)

? ? pose = doc.createElement('pose')

? ? pose.appendChild(doc.createTextNode('Unspecified'))

? ? obj.appendChild(pose)

? ? truncated = doc.createElement('truncated')

? ? truncated.appendChild(doc.createTextNode(str(0)))

? ? obj.appendChild(truncated)

? ? difficult = doc.createElement('difficult')

? ? difficult.appendChild(doc.createTextNode(str(0)))

? ? obj.appendChild(difficult)

? ? bndbox = doc.createElement('bndbox')


? ? xmin = doc.createElement('xmin')

? ? xmin.appendChild(doc.createTextNode(str(datas[1])))

? ? bndbox.appendChild(xmin)


? ? ymin = doc.createElement('ymin')? ? ? ? ? ? ? ?

? ? ymin.appendChild(doc.createTextNode(str(datas[2])))

? ? bndbox.appendChild(ymin)? ? ? ? ? ? ? ?

? ? xmax = doc.createElement('xmax')? ? ? ? ? ? ? ?

? ? xmax.appendChild(doc.createTextNode(str(datas[3])))

? ? bndbox.appendChild(xmax)? ? ? ? ? ? ? ?

? ? ymax = doc.createElement('ymax')? ?

? ? if? '\r' == str(datas[4])[-1] or '\n' == str(datas[4])[-1]:

? ? ? ? data = str(datas[4])[0:-1]

? ? else:

? ? ? ? data = str(datas[4])

? ? ymax.appendChild(doc.createTextNode(data))

? ? bndbox.appendChild(ymax)

? ? obj.appendChild(bndbox)? ? ? ? ? ? ? ?

? ? return obj

def create():

? ? for walk in os.walk(labels):

? ? ? ? for each in walk[2]:

? ? ? ? ? ? fidin=open(walk[0] + '/'+ each,'r')

? ? ? ? ? ? objIndex = 0

? ? ? ? ? ? for data in islice(fidin, 1, None):? ? ? ?

? ? ? ? ? ? ? ? objIndex += 1

? ? ? ? ? ? ? ? data=data.strip('\n')

? ? ? ? ? ? ? ? datas = data.split(' ')

? ? ? ? ? ? ? ? if 5 != len(datas):

? ? ? ? ? ? ? ? ? ? print 'bounding box information error'

? ? ? ? ? ? ? ? ? ? continue

? ? ? ? ? ? ? ? pictureName = each.replace('.txt', '.jpg')

? ? ? ? ? ? ? ? imageFile = imgpath + pictureName

? ? ? ? ? ? ? ? img = cv2.imread(imageFile)

? ? ? ? ? ? ? ? imgSize = img.shape

? ? ? ? ? ? ? ? if 1 == objIndex:

? ? ? ? ? ? ? ? ? ? xmlName = each.replace('.txt', '.xml')

? ? ? ? ? ? ? ? ? ? f = open(xmlpath_new + xmlName, "w")

? ? ? ? ? ? ? ? ? ? doc = Document()

? ? ? ? ? ? ? ? ? ? annotation = doc.createElement('annotation')

? ? ? ? ? ? ? ? ? ? doc.appendChild(annotation)


? ? ? ? ? ? ? ? ? ? folder = doc.createElement('folder')

? ? ? ? ? ? ? ? ? ? folder.appendChild(doc.createTextNode(foldername))

? ? ? ? ? ? ? ? ? ? annotation.appendChild(folder)


? ? ? ? ? ? ? ? ? ? filename = doc.createElement('filename')

? ? ? ? ? ? ? ? ? ? filename.appendChild(doc.createTextNode(pictureName))

? ? ? ? ? ? ? ? ? ? annotation.appendChild(filename)


? ? ? ? ? ? ? ? ? ? source = doc.createElement('source')? ? ? ? ? ? ? ?

? ? ? ? ? ? ? ? ? ? database = doc.createElement('database')

? ? ? ? ? ? ? ? ? ? database.appendChild(doc.createTextNode('My Database'))

? ? ? ? ? ? ? ? ? ? source.appendChild(database)

? ? ? ? ? ? ? ? ? ? source_annotation = doc.createElement('annotation')

? ? ? ? ? ? ? ? ? ? source_annotation.appendChild(doc.createTextNode(foldername))

? ? ? ? ? ? ? ? ? ? source.appendChild(source_annotation)

? ? ? ? ? ? ? ? ? ? image = doc.createElement('image')

? ? ? ? ? ? ? ? ? ? image.appendChild(doc.createTextNode('flickr'))

? ? ? ? ? ? ? ? ? ? source.appendChild(image)

? ? ? ? ? ? ? ? ? ? flickrid = doc.createElement('flickrid')

? ? ? ? ? ? ? ? ? ? flickrid.appendChild(doc.createTextNode('NULL'))

? ? ? ? ? ? ? ? ? ? source.appendChild(flickrid)

? ? ? ? ? ? ? ? ? ? annotation.appendChild(source)


? ? ? ? ? ? ? ? ? ? owner = doc.createElement('owner')

? ? ? ? ? ? ? ? ? ? flickrid = doc.createElement('flickrid')

? ? ? ? ? ? ? ? ? ? flickrid.appendChild(doc.createTextNode('NULL'))

? ? ? ? ? ? ? ? ? ? owner.appendChild(flickrid)

? ? ? ? ? ? ? ? ? ? name = doc.createElement('name')

? ? ? ? ? ? ? ? ? ? name.appendChild(doc.createTextNode('idaneel'))

? ? ? ? ? ? ? ? ? ? owner.appendChild(name)

? ? ? ? ? ? ? ? ? ? annotation.appendChild(owner)


? ? ? ? ? ? ? ? ? ? size = doc.createElement('size')

? ? ? ? ? ? ? ? ? ? width = doc.createElement('width')

? ? ? ? ? ? ? ? ? ? width.appendChild(doc.createTextNode(str(imgSize[1])))

? ? ? ? ? ? ? ? ? ? size.appendChild(width)

? ? ? ? ? ? ? ? ? ? height = doc.createElement('height')

? ? ? ? ? ? ? ? ? ? height.appendChild(doc.createTextNode(str(imgSize[0])))

? ? ? ? ? ? ? ? ? ? size.appendChild(height)

? ? ? ? ? ? ? ? ? ? depth = doc.createElement('depth')

? ? ? ? ? ? ? ? ? ? depth.appendChild(doc.createTextNode(str(imgSize[2])))

? ? ? ? ? ? ? ? ? ? size.appendChild(depth)

? ? ? ? ? ? ? ? ? ? annotation.appendChild(size)


? ? ? ? ? ? ? ? ? ? segmented = doc.createElement('segmented')

? ? ? ? ? ? ? ? ? ? segmented.appendChild(doc.createTextNode(str(0)))

? ? ? ? ? ? ? ? ? ? annotation.appendChild(segmented)? ? ? ? ? ?

? ? ? ? ? ? ? ? ? ? annotation.appendChild(insertObject(doc, datas))

? ? ? ? ? ? ? ? else:

? ? ? ? ? ? ? ? ? ? annotation.appendChild(insertObject(doc, datas))

? ? ? ? ? ? try:

? ? ? ? ? ? ? ? f.write(doc.toprettyxml(indent = '? ? '))

? ? ? ? ? ? ? ? f.close()

? ? ? ? ? ? ? ? fidin.close()

? ? ? ? ? ? except:

? ? ? ? ? ? ? ? pass



if __name__ == '__main__':

? ? create()

新建文件夾，命名為ImageSets，在ImageSets里再新建文件夾，命名為Main。我們可以通過xml名字，生成四個txt文件，即利用createTest.py 生成訓練集和測試集標識文。

main文件夾下會生成test.txt和train.txt，val.txt以及trainval.txt文件。test.txt是測試集，train.txt是訓練集，val.txt是驗證集，trainval.txt是訓練和驗證集.VOC2007中，trainval大概是整個數(shù)據(jù)集的50%，test也大概是整個數(shù)據(jù)集的50%；

train大概是trainval的50%，val大概是trainval的50%

? ? 將生成的JPEGImages，JPEGImages，ImagesSets替換掉VOC2007中的五個文件夾，制作完成。

參考博客：http://www.cnblogs.com/objectDetect/p/5780006.html

? ? http://www.voidcn.com/blog/sinat_30071459/article/p-5745727.html

?著作權(quán)歸作者所有,轉(zhuǎn)載或內(nèi)容合作請聯(lián)系作者
平臺聲明：文章內(nèi)容（如有圖片或視頻亦包括在內(nèi)）由作者上傳并發(fā)布，文章內(nèi)容僅代表作者本人觀點，簡書系信息發(fā)布平臺，僅提供信息存儲服務。
  • 人面猴
    序言：七十年代末，一起剝皮案震驚了整個濱河市，隨后出現(xiàn)的幾起案子，更是在濱河造成了極大的恐慌，老刑警劉巖，帶你破解...
    沈念sama閱讀 228,835評論 6贊 534
  • 死咒
    序言：濱河連續(xù)發(fā)生了三起死亡事件，死亡現(xiàn)場離奇詭異，居然都是意外死亡，警方通過查閱死者的電腦和手機，發(fā)現(xiàn)死者居然都...
    沈念sama閱讀 98,676評論 3贊 419
  • 救了他兩次的神仙讓他今天三更去死
    文/潘曉璐 我一進店門，熙熙樓的掌柜王于貴愁眉苦臉地迎上來，“玉大人，你說我怎么就攤上這事。” “怎么了？”我有些...
    開封第一講書人閱讀 176,730評論 0贊 380
  • 道士緝兇錄：失蹤的賣姜人
    文/不壞的土叔 我叫張陵，是天一觀的道長。 經(jīng)常有香客問我，道長，這世上最難降的妖魔是什么？ 我笑而不...
    開封第一講書人閱讀 63,118評論 1贊 314
  • ?港島之戀（遺憾婚禮）
    正文 為了忘掉前任，我火速辦了婚禮，結(jié)果婚禮上，老公的妹妹穿的比我還像新娘。我一直安慰自己，他們只是感情好，可當我...
    茶點故事閱讀 71,873評論 6贊 410
  • 惡毒庶女頂嫁案：這布局不是一般人想出來的
    文/花漫 我一把揭開白布。 她就那樣靜靜地躺著，像睡著了一般。 火紅的嫁衣襯著肌膚如雪。 梳的紋絲不亂的頭發(fā)上，一...
    開封第一講書人閱讀 55,266評論 1贊 324
  • 城市分裂傳說
    那天，我揣著相機與錄音，去河邊找鬼。 笑死，一個胖子當著我的面吹牛，可吹牛的內(nèi)容都是我干的。 我是一名探鬼主播，決...
    沈念sama閱讀 43,330評論 3贊 443
  • 雙鴛鴦連環(huán)套：你想象不到人心有多黑
    文/蒼蘭香墨 我猛地睜開眼，長吁一口氣：“原來是場噩夢啊……” “哼！你這毒婦竟也來了？” 一聲冷哼從身側(cè)響起，我...
    開封第一講書人閱讀 42,482評論 0贊 289
  • 萬榮殺人案實錄
    序言：老撾萬榮一對情侶失蹤，失蹤者是張志新（化名）和其女友劉穎，沒想到半個月后，有當?shù)厝嗽跇淞掷锇l(fā)現(xiàn)了一具尸體，經(jīng)...
    沈念sama閱讀 49,036評論 1贊 335
  • ?護林員之死
    正文 獨居荒郊野嶺守林人離奇死亡，尸身上長有42處帶血的膿包…… 初始之章·張勛 以下內(nèi)容為張勛視角 年9月15日...
    茶點故事閱讀 40,846評論 3贊 356
  • ?白月光啟示錄
    正文 我和宋清朗相戀三年，在試婚紗的時候發(fā)現(xiàn)自己被綠了。 大學時的朋友給我發(fā)了我未婚夫和他白月光在一起吃飯的照片。...
    茶點故事閱讀 43,025評論 1贊 371
  • 活死人
    序言：一個原本活蹦亂跳的男人離奇死亡，死狀恐怖，靈堂內(nèi)的尸體忽然破棺而出，到底是詐尸還是另有隱情，我是刑警寧澤，帶...
    沈念sama閱讀 38,575評論 5贊 362
  • ?日本核電站爆炸內(nèi)幕
    正文 年R本政府宣布，位于F島的核電站，受9級特大地震影響，放射性物質(zhì)發(fā)生泄漏。R本人自食惡果不足惜，卻給世界環(huán)境...
    茶點故事閱讀 44,279評論 3贊 347
  • 男人毒藥：我在死后第九天來索命
    文/蒙蒙 一、第九天 我趴在偏房一處隱蔽的房頂上張望。 院中可真熱鬧，春花似錦、人聲如沸。這莊子的主人今日做“春日...
    開封第一講書人閱讀 34,684評論 0贊 26
  • 一樁弒父案，背后竟有這般陰謀
    文/蒼蘭香墨 我抬頭看了看天上的太陽。三九已至，卻和暖如春，著一層夾襖步出監(jiān)牢的瞬間，已是汗流浹背。 一陣腳步聲響...
    開封第一講書人閱讀 35,953評論 1贊 289
  • 情欲美人皮
    我被黑心中介騙來泰國打工， 沒想到剛下飛機就差點兒被人妖公主榨干…… 1. 我叫王不留，地道東北人。 一個月前我還...
    沈念sama閱讀 51,751評論 3贊 394
  • 代替公主和親
    正文 我出身青樓，卻偏偏與公主長得像，于是被迫代替她去往敵國和親。 傳聞我的和親對象是個殘疾皇子，可洞房花燭夜當晚...
    茶點故事閱讀 48,016評論 2贊 375

推薦閱讀更多精彩內(nèi)容