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

XML 2020-06-13

转载自：利用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文件标注，格式为：

[html] view plain copy
  1. <annotation>
  2. folder>VOC2012</>
  3. filename>2007_000346.jpgsourcedatabase>TheVOC2007Database>PASCALVOC2007image>flickrsizewidth>500height>375depth>3segmented>1objectname>bottlepose>Unspecifiedtruncated>0difficultbndBoxxmin>124ymin>107xmax>230ymax>343>person>137>78>497>89>202>129>247>Frontal>72>209>111>259>
对应的图片为：



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

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

[cpp] copy
    #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:\\DataSet\\VOCtrainval_11-May-2012\\VOCdevkit\\VOC2012\\JPEGImages\\"+XMLName+".jpg";
  121. XMLName="D:\\DataSet\\VOCtrainval_11-May-2012\\VOCdevkit\\VOC2012\\Annotations\\"+XMLName+".xml";
  122. src=imread(ImgName);
  123. CropImageAccordingToXML(XMLName,src,"person");//根据XML标注文档，从图像src中剪裁出所有person目标，保存为文件
  124. system("pause");
  125. }

源码下载，环境为VS2010 + OpenCV2.4.4 + TinyXML2.6.2

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

编译好的TinyXML2.6.2：

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

相关文章

php输出xml格式字符串
php输出xml格式字符串
J2ME Mobile 3D入门教程系列文章之一
J2ME Mobile 3D入门教程系列文章之一
XML轻松学习手册
XML轻松学习手册
XML入门的常见问题(一)
XML入门的常见问题(一)
XML入门的常见问题(三)
XML入门的常见问题(三)
XML轻松学习手册（2）XML概念
XML轻松学习手册（2）XML概念