前提:
1、待训练图像已经有标签文件xml了(可使用labelImage工具获得)。
2、下载了darknet-master安装包
正文:
在darknet-master/scripts/文件夹下,有voc_label.py文件,是针对VOC数据集生成标签txt文件的,这里把这个文件修改下,用来生成自己数据集的标签txt文件。
修改后的voc_label.py文件如下:
#!/usr/bin/python
# -*- coding:utf-8 -*-
import xml.etree.ElementTree as ET
import pickle
import os
from os import listdir, getcwd
from os.path import join
#classes = ["aeroplane", "bicycle", "bird", "boat", "bottle"]
classes = ["smoke"]#(改!)自己要测的目标类别
def convert(size, box):
dw = 1./(size[0])
dh = 1./(size[1])
x = (box[0] + box[1])/2.0 - 1
y = (box[2] + box[3])/2.0 - 1
w = box[1] - box[0]
h = box[3] - box[2]
x = x*dw
w = w*dw
y = y*dh
h = h*dh
return (x,y,w,h)
def convert_