问题描述:当训练结果出现ValueError: invalid literal for int() with base 10: ‘’,在训练faster rcnn时出现了以上问题。
问题解决方案:这是因为数据的元素,因为xml文件中存在数据为非整数情况,需将里面涉及的代码进行更改,将数据全部变换为整数,以下为批量将xml文件坐标转换为整数的代码
import os
import xml.etree.ElementTree as ET
path = r'E:\xml' # 包含xml的文件夹路径,文件夹名字即可
def edit_xml(xml_file):
"""
修改xml文件
:param xml_file:xml文件的路径
:return:
"""
all_xml_file = os.path.join(path, xml_file)
tree = ET.parse(all_xml_file)
print(all_xml_file)
objs = tree.findall('object')
for ix, obj in enumerate(objs):
obj_bnd = obj.find('bndbox')
obj_xmin = obj_bnd.find('xmin')
obj_ymin = obj_bnd.find('ymin')
obj_xmax = obj_bnd.find('xmax')
obj_ymax = obj_bnd.find('ymax')
xmin = float(obj_xmin.text)
ymin = float(obj_ymin.text)
xmax = float(obj_xmax.text)
ymax = float(obj_ymax.text)
obj_xmin.text = str(round(xmin)) # 四舍五入转为整数
obj_ymin.text = str(round(ymin))
obj_xmax.text = str(round(xmax))
obj_ymax.text = str(round(ymax))
tree.write(all_xml_file, method='xml', encoding='utf-8') # 覆盖更新xml文件
if __name__ == '__main__':
files = os.listdir(path) # 获取文件名列表
for i, file in enumerate(files):
if file.endswith('.xml'):
edit_xml(file)