图像预测理-xml转化为csv文件

àì夳堔傛蜴生んèń 2023-10-11 12:32 8阅读 0赞

import os
    import glob
    import pandas as pd
    import xml.etree.ElementTree as ET
    
    def xml_to_csv(path):
        xml_list = []
        #print(glob.glob(path + '\*.xml'))
        for xml_file in glob.glob(path + '\*.xml'):
            tree = ET.parse(xml_file)
            root = tree.getroot()
            for member in root.findall('object'):
                value = (root.find('filename').text,
                         int(root.find('size')[0].text),
                         int(root.find('size')[1].text),
                         member[0].text,
                         int(member[4][0].text),
                         int(member[4][1].text),
                         int(member[4][2].text),
                         int(member[4][3].text)
                         )
                xml_list.append(value)
        column_name = ['filename', 'width', 'height', 'class', 'xmin', 'ymin', 'xmax', 'ymax']
        xml_df = pd.DataFrame(xml_list, columns=column_name)
        return xml_df
    
    
    def main():
        for image_path in ['train', 'test']:
            xml_df = xml_to_csv(image_path)
            xml_df.to_csv(image_path + '.csv', index=None)
            print('Successfully converted xml to csv.') # There are few records in train file with height & width = 0. So remove it manually.
    
    main()

发表评论取消回复

表情：

评论列表（有 0 条评论，8人围观）

还没有评论，来说两句吧...

相关阅读

相关 matlab 将rgb图像转化为灰度图像

码： close all; %关闭当前所有图形窗口 clear all; %清空工作空间变量 clc; %清屏 x = imread...

爱被打了一巴掌/ 2024年04月19日 10:48/ 0 赞/ 83 阅读

相关图像处理-转化为tfrecords文件

!/usr/bin/env python """ Usage: From tensorflow/models/ Create

忘是亡心i/ 2023年10月11日 12:32/ 0 赞/ 11 阅读

相关图像预测理-xml转化为csv文件

import os import glob import pandas as pd import xml.etree.ElementTree a

àì夳堔傛蜴生んèń/ 2023年10月11日 12:32/ 0 赞/ 9 阅读

相关 .txt文件内容用@符号转化为.csv文件的脚本

你可以使用 Python 来实现将 .txt 文件内容转化为 .csv 文件的脚本。以下是一个例子： import csv 读取 .txt 文件 with

曾经终败给现在/ 2023年09月25日 16:39/ 0 赞/ 20 阅读

相关 Mongo 导出为csv文件

遇到需要从Mongo库导出到csv的情况，特此记录。先贴上在mongo目录下命令行的语句： ./mongoexport -h 10.175.54.77 -u userNa

你的名字/ 2023年06月05日 05:58/ 0 赞/ 14 阅读

相关 0033-labelme格式转化为xml格式

create\_xml\_anno.py -- encoding: utf-8 -- """ @File : create_xml_ann

桃扇骨/ 2023年01月11日 08:49/ 0 赞/ 184 阅读

相关 PHP 读取CSV转化为 UTF-8

CSV文件默认是ANSI,读取里面如果有中文，显示到页面会乱码，可通过如下方法转换。 $handler = fopen($file_path, 'r'); s

缺乏、安全感/ 2022年12月29日 14:13/ 0 赞/ 127 阅读

相关利用OpenCV的imread将RGB图像转化为灰度图像!

通常，我们是利用cvtColor将图像转化为灰度图，但实际上在读取图像的时候便可以将图像转化为灰度图，很简单的操作，只需要将imread的第二个参数置为0即可如下面的代

深藏阁楼爱情的钟/ 2022年08月23日 09:54/ 0 赞/ 239 阅读

相关 php 将xml 数据转化为json

PHP : public function xml_to_json($source) { if(is_file($source)){ //传的是文件

客官°小女子只卖身不卖艺/ 2022年05月22日 04:57/ 0 赞/ 281 阅读

相关将xml文件对象转化为Map对象

一、基本思路先将xml文件转化为document对象，并获取到根节点，然后依次遍历二级节点如果二级节点拥有子节点的话：进入递归如果二级节点没有子节点的话：将数据

逃离我推掉我的手/ 2022年05月14日 09:35/ 0 赞/ 655 阅读