pytorch yolo-v3训练自己的数据集之csv转xml xml文件生成(一)

对于自己数据集需要对其进行labelImg对其进行标注。(这个部分在其他地方会有很多描述)
这里主要叙述将csv数据格式转为xml格式

# .csv-->.xml
# ! /usr/bin/python
# -*- coding:UTF-8 -*-
import os, sys
import glob
from PIL import Image
import csv
import os
import numpy as np
import random
import requests

# VEDAI 图像存储位置
src_img_dir = "F:/小麦比赛/yolov3-master/data/JPEGImages"
# VEDAI 图像的 ground truth 的 txt 文件存放位置
#src_txt_dir = "F:/小麦比赛/yolov3-master/data/labels"
#src_xml_dir = "F:/小麦比赛/yolov3-master/data/labels"
'''
path = "C:/Users/32345/Desktop/YOLOv3_pytorch-master/data/train.csv"
def read_data_csv(path):
    birth_data = []
    with open(path, encoding="utf-8") as csvfile:
        csv_reader = csv.reader(csvfile)  # 使用csv.reader读取csvfile中的文件
        birth_header = next(csv_reader)  # 读取第一行每一列的标题
        for row in csv_reader:  # 将csv 文件中的数据保存到birth_data中
            birth_data.append(row)
    return birth_data
a=read_data_csv(path)
print(a)

'''
img_Lists = glob.glob(src_img_dir + '/*.jpg')
#print(src_img_dir)
# read csv 1 without width and height
file_path = "C:/Users/32345/Desktop/YOLOv3_pytorch-master/data/train.csv"
with open(file_path) as csvfile:
    csv_reader = csv.reader(csvfile)  # 使用csv.reader读取csvfile中的文件
    birth_header = next(csv_reader)  # 读取第一行每一列的标题,birth_header为excel头标
    #print(birth_header)
    #print(csv_reader)
    count = 0
    img_pre = ''
    #bbox = ""
    flag = 0
    #print(row[3])
    for row in csv_reader:  # 将csv 文件中的数据保存到birth_data中
        #print(row[3])
        #len(row[3])
        #print(type(row))
        #first
        #print(row[8])
        if flag == 0:
            img_pre = row[0].split('.')[0]
            #print(img_pre)
            #width = row[1].split('.')[1]
            #print(str(imga_pre))
            flag = 1
            xml_file = open(('F:/小麦比赛/yolov3-master/data/A
  • 3
    点赞
  • 7
    收藏
    觉得还不错? 一键收藏
  • 2
    评论

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

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

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值