1. 自己数据集格式介绍
我们数据集标注的文件如下
每一个image对应一个txt,每个txt内容如下
每一行有8个数据,每个数据代表的含义如下图所示
2. 重命名
VOC2007格式必须为JPG格式,并且图片是统一的六位数字,从000001开始。那我们也需要将所有训练图片重命名
python代码如下:
# -*- coding: utf-8 -*-
import os
path = "E:\\study_materials\\ECCV Vision Meets Drones Challenge\\datasets\\VisDrone2018-DET-train\\VisDrone2018-DET-train\\images_rename"
filelist = os.listdir(path) #该文件夹下所有的文件(包括文件夹)
count=1
for file in filelist:
print(file)
for file in filelist: #遍历所有文件
Olddir=os.path.join(path,file) #原来的文件路径
if os.path.isdir(Olddir): #如果是文件夹则跳过
continue
filename=os.path.splitext(file)[0] #文件名
filetype=os.path.splitext(file)[1] #文件扩展名
Newdir=os.path.join(path,str(count).zfill(6)+filetype) #用字符串函数zfill 以0补全所需位数
os.rename(Olddir,Newdir)#重命名
count+=1
运行代码之后:
变成了:
3. 根据PASCAL VOC数据集制作自己的数据集
我们制作数据集的时候,其实是把我们自己的数据格式转换为PASCAL VOC数据集的格式。
格式如下:
即每行由图片名、目标类型、包围框坐标组成,空格隔开
如果一张图片有多个目标,则格式如下:(比如两个目标)
DG00003L005.jpg iris 187 168 350 253
DG00003L005.jpg iris 232 86 412 168
四个坐标点为包围框坐标的左上角和右下角。
所以现在要做的就是把txt文件中的信息先变成如上图所示的格式
小处理:
- voc2007里面的xml,没有occlusion这个参数,所以暂时忽略。
- 并且把第五个参数 score为0的都跳过,不考虑。
- 此比赛存储的是左上角和长宽信息,所以需要转化成左上角和右下角信息。
- 网上默认xml里面的truncation参数都为0,所以txt里面没有此参数,而我这边生成的txt文件中加入了truncation参数,而且后面matlab转换xml代码中也多了一个str{7}。
转化代码如下:
#include <fstream>
#include <sstream>
#include <iostream>
#include <io.h>
#include <vector>
using namespace std;
const string baseFilePath = "E:\\study_materials\\ECCV Vision Meets Drones Challenge\\datasets\\VisDrone2018-DET-train\\VisDrone2018-DET-train\\annotations1\\";
/************************************************************************/
/* 获取文件夹下所有文件名
输入:
path : 文件夹路径
exd : 所要获取的文件名后缀,如jpg、png等;如果希望获取所有
文件名, exd = ""
输出:
files : 获取的文件名列表
*/
/************************************************************************/
void getFiles(string path, string exd, vector<string>& files)
{
//文件句柄
long hFile = 0;
//文件信息
struct _finddata_t fileinfo;
string pathName, exdName;
if (0 != strcmp(exd.c_str(), ""))
{
exdName = "\\*." + exd;
}
else
{
exdName = "\\*";
}