利用TinyXML读取VOC2012数据集的XML标注文件裁剪出所有人体目标

转载自：利用TinyXML读取VOC2012数据集的XML标注文件裁剪出所有人体目标保存为文件 - Why So Serious? - 博客频道 - CSDN.NET http://blog.csdn.net/masibuaa/article/details/16104717

PASCAL VOC目标检测数据集(The PASCAL Visual Object Classes)

http://pascallin.ecs.soton.ac.uk/challenges/VOC/

图片中的目标用XML文件标注，格式为：

所以如果想用这个数据集做某种目标识别的训练集的话，需要先从中裁出需要的目标。

下面这个程序就是这个目的，其中用到了TinyXML这个简单易用的XML解析器(XML入门)

#include<iostream>
1. #include<fstream>
2. #include<opencv2/core/core.hpp>
3. #include<opencv2/highgui/highgui.hpp>
4. #include<opencv2/imgproc/imgproc.hpp>
5. #include<opencv2/objdetect/objdetect.hpp>
6. #include<opencv2/ml/ml.hpp>
8. #include<tinyxml.h>
9. usingnamespacestd;
10. namespacecv;
12. intCropImageCount=0;//裁剪出来的人体图片个数
13. /**
14. *通过根节点和节点名查找所有指定节点，结果放到节点数组NodeVector中
15. *@parampRootElexml文件的根节点
16. *@paramstrNodeName要查询的节点名
17. *@paramNodeVector查询到的节点指针数组
18. *@return找到至少一个相应节点，返回true；否则false
19. */
20. boolGetAllNodePointerByName(TiXmlElement*pRootEle,stringstrNodeName,vector<TiXmlElement*>&NodeVector)
21. {
22. //如果NodeName等于根节点名，加入NodeVector数组
23. if(strNodeName==pRootEle->Value())
24. {
25. NodeVector.push_back(pRootEle);//添加到数组末尾
26. //这里根据VOCAnnotation的XML文件格式，认为相同节点名的节点不会有父子关系，所以所有相同节点名的节点都在同一级别上
27. //只要找到第一个，剩下的肯定在它的兄弟节点里面
28. for(TiXmlElement*pElement=pRootEle->NextSiblingElement();pElement;pElement=pElement->NextSiblingElement())
29. if(strNodeName==pElement->Value())
30. NodeVector.push_back(pElement);
31. returntrue;
32. }
33. TiXmlElement*pEle=pRootEle;
34. for(pEle=pRootEle->FirstChildElement();pEle;pEle=pEle->NextSiblingElement())
35. //递归处理子节点，获取节点指针
36. if(GetAllNodePointerByName(pEle,strNodeName,NodeVector))
37. true;
38. }
39. false;//没找到
40. *根据目标名过滤目标节点数组,删除所有目标名不是objectName的元素
41. *@paramNodeVector要操作的TiXmlElement元素指针数组
42. *@paramobjectName指定的目标名，删除所有目标名不是objectName的元素
43. *@return过滤后目标数组为空，返回false；否则返回true
44. */
45. boolFiltObject(vector<TiXmlElement*>&NodeVector,stringobjectName)
46. TiXmlElement*pEle=NULL;
47. vector<TiXmlElement*>::iteratoriter=NodeVector.begin();//数组的迭代器
48. for(;iter!=NodeVector.end();)
49. pEle=*iter;//第i个元素
50. //若目标名不是objectName，删除此节点
51. if(objectName!=pEle->FirstChildElement()->GetText())
52. //cout<<"删除的目标节点："<<pEle->FirstChildElement()->GetText()<<endl;
53. iter=NodeVector.erase(iter);//删除目标名不是objectName的，返回下一个元素的指针
54. else
55. iter++;
56. if(0==NodeVector.size())//过滤后目标数组为空，说明不包含指定目标
57. false;
58. else
59. *根据每个目标的BoundingBox，剪裁图像，保存为文件
60. *@paramimg图像
61. *@paramNodeVector目标节点数组
62. voidCropImage(Matimg,vector<TiXmlElement*>NodeVector)
63. intxmin,ymin,xmax,ymax;//从目标节点中读出的包围盒参数
64. charfileName[256];//剪裁后的图片和其水平翻转图片的文件名
65. //遍历目标数组
66. for(;iter!=NodeVector.end();iter++)
67. //遍历每个目标的子节点
68. TiXmlElement*pEle=(*iter)->FirstChildElement();//第i个元素的第一个孩子
69. for(;pEle;pEle=pEle->NextSiblingElement())
70. //找到包围盒"bndbox"节点
71. if(string("bndbox")==pEle->Value())
72. TiXmlElement*pCoord=pEle->FirstChildElement();//包围盒的第一个坐标值
73. //依次遍历包围盒的4个坐标值，放入整型变量中
74. for(;pCoord;pCoord=pCoord->NextSiblingElement())
75. if(string("xmin")==pCoord->Value())
76. xmin=atoi(pCoord->GetText());//xmin
77. if(string("ymin")==pCoord->Value())
78. ymin=atoi(pCoord->GetText());//ymin
79. if(string("xmax")==pCoord->Value())
80. xmax=atoi(pCoord->GetText());//xmax
81. if(string("ymax")==pCoord->Value())
82. ymax=atoi(pCoord->GetText());//ymax
83. //cout<<"xmin:"<<xmin<<","<<"ymin:"<<ymin<<","<<"xmax:"<<xmax<<","<<"ymax:"<<ymax<<endl;;
84. //根据读取的包围盒坐标设置图像ROI
85. MatimgROI=img(Rect(xmin,xmax-xmin,ymax-ymin));
86. resize(imgROI,imgROI,Size(64,128));//缩放为64*128大小
87. sprintf(fileName,"person%06d.jpg",++CropImageCount);//生成剪裁图片的文件名
88. imwrite(fileName,imgROI);//保存文件
89. flip(imgROI,1);//水平翻转
90. memset(fileName,0x00,sizeof(fileName));
91. //生成剪裁图片的水平翻转图片的文件名
92. /**
93. *根据XML文件，从图像中剪裁出objectName目标
94. *@paramXMLFileXML文件名
95. *@paramimg对应的图像
96. *@paramobjectName目标名
97. *@return若图像中包含objectName目标，返回true；否则返回false
98. boolCropImageAccordingToXML(stringXMLFile,Matimg,248)"> TiXmlDocument*pDoc=newTiXmlDocument();//创建XML文档
99. pDoc->LoadFile(XMLFile.c_str());//装载XML文件
100. vector<TiXmlElement*>nodeVector;//节点数组
101. //查找所有节点名是object的节点，即目标节点，结果放到节点数组nodeVector中
102. if(false==GetAllNodePointerByName(pDoc->RootElement(),"object",nodeVector))//未找到指定目标
103. false;
104. //cout<<"所有目标个数："<<nodeVector.size()<<endl;
105. //过滤节点数组，删除所有节点名不是objectName的节点
106. false==FiltObject(nodeVector,objectName))//目标数组中没有指定目标
107. //cout<<"过滤后的目标个数："<<nodeVector.size()<<endl;
108. //根据每个目标的BoundingBox，剪裁图像，保存为文件
109. CropImage(img,nodeVector);
110. intmain()
111. intfileCount=0;//文件个数
112. Matsrc;
113. stringXMLName,ImgName;//XML文件名和对应的图片文件名
114. //ifstreamfin("VOC2012AnnotationsXMLList.txt");//打开XML文件列表
115. ifstreamfin("subset.txt");
116. //ifstreamfin("test.txt");
117. //读取XML文件列表
118. while(getline(fin,XMLName))
119. cout<<"处理："<<XMLName<<endl;
120. ImgName="D:DataSetVOCtrainval_11-May-2012VOCdevkitVOC2012JPEGImages"+XMLName+".jpg";
121. XMLName="D:DataSetVOCtrainval_11-May-2012VOCdevkitVOC2012Annotations"+XMLName+".xml";
122. src=imread(ImgName);
123. CropImageAccordingToXML(XMLName,src,"person");//根据XML标注文档，从图像src中剪裁出所有person目标，保存为文件
124. system("pause");
125. }

http://download.csdn.net/detail/masikkk/6547823

编译好的TinyXML2.6.2：

http://download.csdn.net/detail/masikkk/6547809

（编辑：李大同）

【声明】本站内容均来自网络，其相关言论仅代表作者个人观点，不代表本站立场。若无意侵犯到您的权利，请及时与联系站长删除相关内容!