• 目标检测数据集格式转换：txt格式转换为xml格式（以VisDrone数据集为例）

    1.准备好两个文件夹

    VisDroneTxt文件夹里面装的是原图片以及txt格式的标签

    VisDroneVoc里面的labels文件夹是目标文件夹，用来装转换之后的xml格式标签

    2.给出原转换程序

    1. # .txt-->.xml
    2. # ! /usr/bin/python
    3. # -*- coding:UTF-8 -*-
    4. import os
    5. import cv2
    6.  
    7.  
    8. def txt_to_xml(txt_path, img_path, xml_path):
    9.     # 1.字典对标签中的类别进行转换
    10.     dict = {'0': "organoid",
    11.             '1': "car",
    12.             '2': "bus",
    13.             '3': "ufo",
    14.             '4': "robot",
    15.             '5': "virus",
    16.             '6': "trunk",
    17.             '7': "plash",
    18.             '8': "biycle"}
    19.     # 2.找到txt标签文件夹
    20.     files = os.listdir(txt_path)
    21.     # 用于存储 "老图"
    22.     pre_img_name = ''
    23.     # 3.遍历文件夹
    24.     for i, name in enumerate(files):
    25.         # 许多人文件夹里有该文件，默认的也删不掉，那就直接pass
    26.         if name == "desktop.ini":
    27.             continue
    28.         print(name)
    29.         # 4.打开txt
    30.         txtFile = open(txt_path + name)
    31.         # 读取所有内容
    32.         txtList = txtFile.readlines()
    33.         # 读取图片名称
    34.         img_name = name.split(".")[0]
    35.         pic = cv2.imread(img_path + img_name + ".jpg")
    36.         # 获取图像大小信息
    37.         Pheight, Pwidth, Pdepth = pic.shape
    38.         # 5.遍历txt文件中每行内容
    39.         for row in txtList:
    40.             # 按' '分割txt的一行的内容
    41.             oneline = row.strip().split(" ")
    42.             # 遇到的是一张新图片
    43.             if img_name != pre_img_name:
    44.                 # 6.新建xml文件
    45.                 xml_file = open((xml_path + img_name + '.xml'), 'w')
    46.                 xml_file.write('\n')
    47.                 xml_file.write('    VOC2007\n')
    48.                 xml_file.write('    ' + img_name + '.jpg' + '\n')
    49.                 xml_file.write('\n')
    50.                 xml_file.write('orgaquant\n')
    51.                 xml_file.write('organoids\n')
    52.                 xml_file.write('\n')
    53.                 xml_file.write('    \n')
    54.                 xml_file.write('        ' + str(Pwidth) + '\n')
    55.                 xml_file.write('        ' + str(Pheight) + '\n')
    56.                 xml_file.write('        ' + str(Pdepth) + '\n')
    57.                 xml_file.write('    \n')
    58.                 xml_file.write('    \n')
    59.                 xml_file.write('' + dict[oneline[0]] + '\n')
    60.                 xml_file.write('        \n')
    61.                 xml_file.write('            ' + str(
    62.                     int(((float(oneline[1])) * Pwidth + 1) - (float(oneline[3])) * 0.5 * Pwidth)) + '\n')
    63.                 xml_file.write('            ' + str(
    64.                     int(((float(oneline[2])) * Pheight + 1) - (float(oneline[4])) * 0.5 * Pheight)) + '\n')
    65.                 xml_file.write('            ' + str(
    66.                     int(((float(oneline[1])) * Pwidth + 1) + (float(oneline[3])) * 0.5 * Pwidth)) + '\n')
    67.                 xml_file.write('            ' + str(
    68.                     int(((float(oneline[2])) * Pheight + 1) + (float(oneline[4])) * 0.5 * Pheight)) + '\n')
    69.                 xml_file.write('        \n')
    70.                 xml_file.write('    
      71. \n'      )
    71.                 xml_file.close()
    72.                 pre_img_name = img_name  # 将其设为"老"图
    73.             else:  # 不是新图而是"老图"
    74.                 # 7.同一张图片，只需要追加写入object
    75.                 xml_file = open((xml_path + img_name + '.xml'), 'a')
    76.                 xml_file.write('    \n')
    77.                 xml_file.write('' + dict[oneline[0]] + '\n')
    78.                 '''  按需添加这里和上面
    79.                 xml_file.write('        Unspecified\n')
    80.                 xml_file.write('        0\n')
    81.                 xml_file.write('        0\n')
    82.                 '''
    83.                 xml_file.write('        \n')
    84.                 xml_file.write('            ' + str(
    85.                     int(((float(oneline[1])) * Pwidth + 1) - (float(oneline[3])) * 0.5 * Pwidth)) + '\n')
    86.                 xml_file.write('            ' + str(
    87.                     int(((float(oneline[2])) * Pheight + 1) - (float(oneline[4])) * 0.5 * Pheight)) + '\n')
    88.                 xml_file.write('            ' + str(
    89.                     int(((float(oneline[1])) * Pwidth + 1) + (float(oneline[3])) * 0.5 * Pwidth)) + '\n')
    90.                 xml_file.write('            ' + str(
    91.                     int(((float(oneline[2])) * Pheight + 1) + (float(oneline[4])) * 0.5 * Pheight)) + '\n')
    92.                 xml_file.write('        \n')
    93.                 xml_file.write('    
      94. \n'      )
    94.                 xml_file.close()
    95.  
    96.         # 8.读完txt文件最后写入
    97.         xml_file1 = open((xml_path + pre_img_name + '.xml'), 'a')
    98.         xml_file1.write('')
    99.         xml_file1.close()
    100.     print("Done !")
    101.  
    102.  
    103. # 修改成自己的文件夹 注意文件夹最后要加上/
    104. txt_to_xml("txt_path/", "img_path/", "xml_path/")

    3.需要修改的地方

    3.1 修改标签类别：

    修改之后

    3.2将路径修改成自己文件夹的路径

    3.3 修改之后的完整代码

    1. # .txt-->.xml
    2. # ! /usr/bin/python
    3. # -*- coding:UTF-8 -*-
    4. import os
    5. import cv2
    6.  
    7.  
    8. def txt_to_xml(txt_path, img_path, xml_path):
    9.     # 1.字典对标签中的类别进行转换
    10.     dict = {'0': "pedestrian",
    11.             '1': "people",
    12.             '2': "bicycle",
    13.             '3': "car",
    14.             '4': "van",
    15.             '5': "truck",
    16.             '6': "tricycle",
    17.             '7': "awning-tricycle",
    18.             '8': "bus",
    19.             '9': "motor"}
    20.     # 2.找到txt标签文件夹
    21.     files = os.listdir(txt_path)
    22.     # 用于存储 "老图"
    23.     pre_img_name = ''
    24.     # 3.遍历文件夹
    25.     for i, name in enumerate(files):
    26.         # 许多人文件夹里有该文件，默认的也删不掉，那就直接pass
    27.         if name == "desktop.ini":
    28.             continue
    29.         print(name)
    30.         # 4.打开txt
    31.         txtFile = open(txt_path + name)
    32.         # 读取所有内容
    33.         txtList = txtFile.readlines()
    34.         # 读取图片名称
    35.         img_name = name.split(".")[0]
    36.         pic = cv2.imread(img_path + img_name + ".jpg")
    37.         # 获取图像大小信息
    38.         Pheight, Pwidth, Pdepth = pic.shape
    39.         # 5.遍历txt文件中每行内容
    40.         for row in txtList:
    41.             # 按' '分割txt的一行的内容
    42.             oneline = row.strip().split(" ")
    43.             # 遇到的是一张新图片
    44.             if img_name != pre_img_name:
    45.                 # 6.新建xml文件
    46.                 xml_file = open((xml_path + img_name + '.xml'), 'w')
    47.                 xml_file.write('\n')
    48.                 xml_file.write('    VOC2007\n')
    49.                 xml_file.write('    ' + img_name + '.jpg' + '\n')
    50.                 xml_file.write('\n')
    51.                 xml_file.write('orgaquant\n')
    52.                 xml_file.write('organoids\n')
    53.                 xml_file.write('\n')
    54.                 xml_file.write('    \n')
    55.                 xml_file.write('        ' + str(Pwidth) + '\n')
    56.                 xml_file.write('        ' + str(Pheight) + '\n')
    57.                 xml_file.write('        ' + str(Pdepth) + '\n')
    58.                 xml_file.write('    \n')
    59.                 xml_file.write('    \n')
    60.                 xml_file.write('' + dict[oneline[0]] + '\n')
    61.                 xml_file.write('        \n')
    62.                 xml_file.write('            ' + str(
    63.                     int(((float(oneline[1])) * Pwidth + 1) - (float(oneline[3])) * 0.5 * Pwidth)) + '\n')
    64.                 xml_file.write('            ' + str(
    65.                     int(((float(oneline[2])) * Pheight + 1) - (float(oneline[4])) * 0.5 * Pheight)) + '\n')
    66.                 xml_file.write('            ' + str(
    67.                     int(((float(oneline[1])) * Pwidth + 1) + (float(oneline[3])) * 0.5 * Pwidth)) + '\n')
    68.                 xml_file.write('            ' + str(
    69.                     int(((float(oneline[2])) * Pheight + 1) + (float(oneline[4])) * 0.5 * Pheight)) + '\n')
    70.                 xml_file.write('        \n')
    71.                 xml_file.write('    
      72. \n'      )
    72.                 xml_file.close()
    73.                 pre_img_name = img_name  # 将其设为"老"图
    74.             else:  # 不是新图而是"老图"
    75.                 # 7.同一张图片，只需要追加写入object
    76.                 xml_file = open((xml_path + img_name + '.xml'), 'a')
    77.                 xml_file.write('    \n')
    78.                 xml_file.write('' + dict[oneline[0]] + '\n')
    79.                 '''  按需添加这里和上面
    80.                 xml_file.write('        Unspecified\n')
    81.                 xml_file.write('        0\n')
    82.                 xml_file.write('        0\n')
    83.                 '''
    84.                 xml_file.write('        \n')
    85.                 xml_file.write('            ' + str(
    86.                     int(((float(oneline[1])) * Pwidth + 1) - (float(oneline[3])) * 0.5 * Pwidth)) + '\n')
    87.                 xml_file.write('            ' + str(
    88.                     int(((float(oneline[2])) * Pheight + 1) - (float(oneline[4])) * 0.5 * Pheight)) + '\n')
    89.                 xml_file.write('            ' + str(
    90.                     int(((float(oneline[1])) * Pwidth + 1) + (float(oneline[3])) * 0.5 * Pwidth)) + '\n')
    91.                 xml_file.write('            ' + str(
    92.                     int(((float(oneline[2])) * Pheight + 1) + (float(oneline[4])) * 0.5 * Pheight)) + '\n')
    93.                 xml_file.write('        \n')
    94.                 xml_file.write('    
      95. \n'      )
    95.                 xml_file.close()
    96.  
    97.         # 8.读完txt文件最后写入
    98.         xml_file1 = open((xml_path + pre_img_name + '.xml'), 'a')
    99.         xml_file1.write('')
    100.         xml_file1.close()
    101.     print("Done !")
    102.  
    103.  
    104. # 修改成自己的文件夹 注意文件夹最后要加上/
    105. txt_to_xml("VisDroneTxt/labels/", "VisDroneTxt/images/", "VisDroneXml/labels/")

    4.运行.py文件

    5.转换成功

    到此为止，将Visdrone数据集的标签格式从txt转换为xml。

  • 相关阅读:
    spring 中 @Validated/@Valid
    LeetCode 1704. 判断字符串的两半是否相似
    跨平台开发新纪元：Xcode的多平台应用构建指南
    本科生学深度学习，搭建环境，再不入坑就晚了
    HCIP路由交换的三门考试
    C与CPP常见编译工具链与构建系统简介
    Mysql 日常命令记录
    虚拟机中Linux下安装服务器
    【obs】x264编码：“buffer_size“
    8月8本程序员新书：Rust、软技能、Python、量子...
  • 原文地址：https://blog.csdn.net/pyscl01/article/details/133745965