yolo中的txt转换成需要的xml

#include<iostream> #include <fstream> #include<string> #include <sstream> #include<vector> //打开xml文件需要加载的头文件 #include "tinystr.h" #include "tinyxml.h" using namespace std; struct WHSize { int width; int height; }; //box参数结构体 struct BoxSize { int xMin; int yMin; int xMax; int yMax; }; void writeFile(string &address, vector<vector<double>>&data) { ofstream fout(address); if (!fout.is_open()) { cout << "open file error!" << endl; } for (int i = 0; i < data.size(); i++) { for (int k = 0; k < data[i].size(); k++) { fout << data[i][k] << " "; } fout << endl; } data.clear(); fout.close(); } bool txt2xml(string file_txt, string file_xml, string file_names) //file_names类型文件路径 file_txt 文件 转换file_xml路径 { vector<string>stringnames; string TypeNamePath = file_names; ifstream namesin; namesin.open(TypeNamePath, ios::in);//打开一个file_txt if (!namesin.is_open()) { cout << "Can not find target file." << endl; return -1; } string namesBuff; while (!namesin.eof()) { namesin >> namesBuff; if (namesin.fail()) { break; } stringnames.push_back(namesBuff);//存储类型vector } namesin.close(); string FilePath = file_txt; ifstream in; in.open(FilePath, ios::in);//打开一个file_txt if (!in.is_open()) { cout << "Can not find target file." << endl; return -1; } string buff;//按空格读取数据string类型 int num = 0; vector<double>nvec; vector<vector<double>>nresult; while (!in.eof()) { in >> buff; double n = atof(buff.c_str()); cout << n << endl; nvec.push_back(n); num++; if (num % 5 == 0) { nresult.push_back(nvec);//二维数组保存数据 nvec.clear(); } } in.close(); //x y w h //公式 s= x*2000*2 z= w*2000 xmax=(s+z)/2 xmin=xmax-z //同理 s=y*2000*2 z=h*2000 ymax=(s+z)/2 ymin=ymax-z //确定有几个object int n = nresult.size(); // const char * xmlFile = file_xml.c_str(); TiXmlDocument *doc = new TiXmlDocument; //xml文档指针 //文档格式声明 TiXmlDeclaration *decl = new TiXmlDeclaration("1.0", "", ""); doc->LinkEndChild(decl); //写入文档 TiXmlElement *RootElement = new TiXmlElement("annotation");//根元素 // TiXmlElement *folderElement = new TiXmlElement("folder");//图片在的文件夹 固定 RootElement->LinkEndChild(folderElement); TiXmlText *folderContent = new TiXmlText("datas"); folderElement->LinkEndChild(folderContent); // TiXmlElement *filenameElement = new TiXmlElement("filename");//图片的名字 循环读取 RootElement->LinkEndChild(filenameElement); TiXmlText *filenameContent = new TiXmlText("0.jpg"); filenameElement->LinkEndChild(filenameContent); // TiXmlElement *pathElement = new TiXmlElement("path"); RootElement->LinkEndChild(pathElement); TiXmlText *pathContent = new TiXmlText("D:\\datas\\0.jpg");//不能含有中文 循环读取 pathElement->LinkEndChild(pathContent); // TiXmlElement *sourceElement = new TiXmlElement("source");//source // TiXmlElement *databaseElement = new TiXmlElement("database"); sourceElement->LinkEndChild(databaseElement); TiXmlText *databaseContent = new TiXmlText("Unknown");//不能含有中文 databaseElement->LinkEndChild(databaseContent); // TiXmlElement *sizeElement = new TiXmlElement("size");//size // TiXmlElement *widthElement = new TiXmlElement("width"); sizeElement->LinkEndChild(widthElement); TiXmlText *widthContent = new TiXmlText("2000");//不能含有中文 widthElement->LinkEndChild(widthContent); // TiXmlElement *heightElement = new TiXmlElement("height"); sizeElement->LinkEndChild(heightElement); TiXmlText *heightContent = new TiXmlText("2000");//不能含有中文 heightElement->LinkEndChild(heightContent); // TiXmlElement *depthElement = new TiXmlElement("depth"); sizeElement->LinkEndChild(depthElement); TiXmlText *depthContent = new TiXmlText("1");//不能含有中文 depthElement->LinkEndChild(depthContent); // TiXmlElement *segmentedElement = new TiXmlElement("segmented"); TiXmlText *segmentedContent = new TiXmlText("0"); segmentedElement->LinkEndChild(segmentedContent); // RootElement->LinkEndChild(sourceElement);//父节点写入文档 RootElement->LinkEndChild(sizeElement);//父节点写入文档 RootElement->LinkEndChild(segmentedElement);//父节点写入文档 // for (int i = 0; i < n; i++) { //读取nresult 转换数据->xml //type x y w h // 0 1 2 3 4 //公式 s= x*2000*2 z= w*2000 xmax=(s+z)/2 xmin=xmax-z //同理 s=y*2000*2 z=h*2000 ymax=(s+z)/2 ymin=ymax-z string type = stringnames[nresult[i][0]]; int sx = nresult[i][1]*2000*2;//x int zw = nresult[i][3]*2000;//w string xmaxval = to_string((sx + zw) / 2); string xminval = to_string((sx + zw) / 2 - zw); int sy = nresult[i][2] * 2000 * 2;//x int zh = nresult[i][4] * 2000;//w string ymaxval = to_string((sy + zh) / 2); string yminval = to_string((sy + zh) / 2 - zh); // TiXmlElement *objectElement = new TiXmlElement("object");//object 有几个类型就有几个object // TiXmlElement *nameElement = new TiXmlElement("name"); objectElement->LinkEndChild(nameElement); TiXmlText *nameContent = new TiXmlText(type.c_str());//不能含有中文 类型 判断读取 nameElement->LinkEndChild(nameContent); // TiXmlElement *poseElement = new TiXmlElement("pose"); objectElement->LinkEndChild(poseElement); TiXmlText *poseContent = new TiXmlText("Unspecified");//不能含有中文 poseElement->LinkEndChild(poseContent); // TiXmlElement *truncatedElement = new TiXmlElement("truncated"); objectElement->LinkEndChild(truncatedElement); TiXmlText *truncatedContent = new TiXmlText("0");//不能含有中文 truncatedElement->LinkEndChild(truncatedContent); // TiXmlElement *difficultElement = new TiXmlElement("difficult"); objectElement->LinkEndChild(difficultElement); TiXmlText *difficultContent = new TiXmlText("0");//不能含有中文 difficultElement->LinkEndChild(difficultContent); // TiXmlElement *bndboxElement = new TiXmlElement("bndbox");//size 坐标信息 计算后循环写入 // TiXmlElement *xminElement = new TiXmlElement("xmin"); bndboxElement->LinkEndChild(xminElement); TiXmlText *xminContent = new TiXmlText(xminval.c_str());//不能含有中文 xminElement->LinkEndChild(xminContent); // TiXmlElement *yminElement = new TiXmlElement("ymin"); bndboxElement->LinkEndChild(yminElement); TiXmlText *yminContent = new TiXmlText(yminval.c_str());//不能含有中文 yminElement->LinkEndChild(yminContent); // TiXmlElement *xmaxElement = new TiXmlElement("xmax"); bndboxElement->LinkEndChild(xmaxElement); TiXmlText *xmaxContent = new TiXmlText(xmaxval.c_str());//不能含有中文 xmaxElement->LinkEndChild(xmaxContent); // TiXmlElement *ymaxElement = new TiXmlElement("ymax"); bndboxElement->LinkEndChild(ymaxElement); TiXmlText *ymaxContent = new TiXmlText(ymaxval.c_str());//不能含有中文 ymaxElement->LinkEndChild(ymaxContent); // objectElement->LinkEndChild(bndboxElement); RootElement->LinkEndChild(objectElement);//父节点写入文档 } doc->LinkEndChild(RootElement); doc->SaveFile(xmlFile); return true; } int main() { vector<string>stringnames; string TypeNamePath = "D:\\ZZ\\15.jpg"; int pos0 = TypeNamePath.rfind("\\"); int length = pos0 - 0; std::string tyimg = TypeNamePath.substr(pos0 + 1, TypeNamePath.length()); string rest= TypeNamePath.substr(0, length); int pos1 = rest.rfind("\\