VOC转COCO数据集（图片重命名、重分配）（傻瓜一站式教程）

1.修改文件名

import os.path

def rename(img_folder,num):
    for img_name in os.listdir(img_folder):  # os.listdir()： 列出路径下所有的文件
        #os.path.join() 拼接文件路径
        src = os.path.join(img_folder, img_name)   #src：要修改的目录名
        dst = os.path.join(img_folder, str(num) + '.xml') #dst： 修改后的目录名      注意此处str(num)将num转化为字符串,继而拼接
        num= num+1
        os.rename(src, dst) #用dst替代src


def main():
    img_folder0 = 'E:\Temp\sirst-master\coco_type\VOC\Annotations' #图片的文件夹路径    直接放你的文件夹路径即可
    num=1
    rename(img_folder0,num)

if __name__=="__main__":
    main()

2.修改xml内容的filename

import xml.dom.minidom
import os

path = r'F:\Pycharm Community\foods\xml1'  # xml文件存放路径
sv_path = r'F:\Pycharm Community\foods\xml2'  # 修改后的xml文件存放路径
files = os.listdir(path)

for xmlFile in files:
    dom = xml.dom.minidom.parse(os.path.join(path, xmlFile))  # 打开xml文件，送到dom解析
    root = dom.documentElement  # 得到文档元素对象
    names = root.getElementsByTagName('filename')
    a, b = os.path.splitext(xmlFile)  # 分离出文件名a
    for n in names:
        n.firstChild.data = a + '.jpg'
    with open(os.path.join(sv_path, xmlFile), 'w') as fh:
        dom.writexml(fh)

3.voc2coco
参考教程