python IndexError:child index out of range

使用yolo进行数据训练将xml文件转换为yolo所需的归一化txt文件时,出现以下错误:

在这里插入图片描述

xml转换为txt文件代码:

import glob
import xml.etree.ElementTree as ET

#  类名
class_names = ['helmet','person']
#  xml文件路径,train_images只需改为val_images就可以处理val_images的了
path = r''  # xml文件路径
save_path = r''  # 归一化后txt文件保存的路径


#  转换一个xml文件为txt
def single_xml_to_txt(xml_file):
    tree = ET.parse(xml_file)
    root = tree.getroot()
    # 保存的txt文件路径
    tmp = xml_file.split('\\')[-1]
    num = tmp.split('.')[0]
    print("fileName = ", num)

    # txt_file = path + 'txtfile' + xml_file.split('.')[0] + '.txt'
    txt_file = save_path +  num + '.txt'
    with open(txt_file, 'w') as txt_file:
        for member in root.findall('object'):
            # filename = root.find('filename').text
            picture_width = int(root.find('size')[0].text)
            picture_height = int(root.find('size')[1].text)
            class_name = member[0].text
            #  类名对应的index
            class_num = class_names.index(class_name)
            box_x_min = int(member[4][0].text)  # 左上角横坐标
            box_y_min = int(member[4][1].text)  # 左上角纵坐标
            box_x_max = int(member[4][2].text)  # 右下角横坐标
            box_y_max = int(member[4][3].text)  # 右下角纵坐标
            # 转成相对位置和宽高
            x_center = (box_x_min + box_x_max) / (2 * picture_width)
            y_center = (box_y_min + box_y_max) / (2 * picture_height)
            width = (box_x_max - box_x_min) / picture_width
            height = (box_y_max - box_y_min) / picture_height
            print(class_num, x_center, y_center, width, height)
            txt_file.write(str(class_num) + ' ' + str(x_center) + ' ' + str(y_center) + ' ' + str(width) + ' ' + str(
                height) + '\n')


#  转换文件夹下的所有xml文件为txt
def dir_xml_to_txt(path):
    for xml_file in glob.glob(path + '*.xml'):
        single_xml_to_txt(xml_file)


dir_xml_to_txt(path)

报错原因:
打开xml文件:
在这里插入图片描述
可以看到bndbox是object下的第5个标签(从第0开始算),而不是第4个(代码中写的是4)

解决办法:
修改代码为:

box_x_min = int(member[5][0].text)  # 左上角横坐标
box_y_min = int(member[5][1].text)  # 左上角纵坐标
box_x_max = int(member[5][2].text)  # 右下角横坐标
box_y_max = int(member[5][3].text)  # 右下角纵坐标
  • 5
    点赞
  • 4
    收藏
    觉得还不错? 一键收藏
  • 1
    评论

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论 1
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值