csv格式转xml文件

csv数据标注格式转xml文件格式，便于SSD和YOLO训练 详见：https://blog.csdn.net/Mynameisyournamewuyu/article/details/81124692

# encoding: utf-8
import csv
import cv2

newRows=[]
with open（“./train_labels.csv“ “r“） as fr:
    rows = csv.reader（fr）
    aRow=[]
    lastJpg=‘‘
    newJpg=‘‘
    count=0
    for row in rows:
        newJpg=row[0]
        if newJpg!=lastJpg:
            if aRow:
                newRows.append（aRow）
            aRow=row
        else:
            aRow.append（row[1]）   
        lastJpg=row[0]
        count+=1
    newRows.append（aRow）            



def getxml（jpgDetail）:
    context=“\n“
    context+=“\tJPGImage\n“
    context+=“\t“+jpgDetail[0]+“\n“
    context+=“\t/MYPATH/JPGImage/“+jpgDetail[0]+“\n“
    context+=“\t\n“
    context+=“\t\tse>Unknownse>\n“
    context+=“\t\n“
    context+=“\t\n“
    img = cv2.imread（“./jpg/“+jpgDetail[0]）#返回一个Image对象
    sp=img.shape
    context+=“\t\t“+str（sp[1]）+“\n“
    context+=“\t\t“+str（sp[0]）+“\n“
    context+=“\t\t