PASCAL VOC转换为COCO格式的脚本

找了很多的脚本，但是不能一次性完成json文件的生成和图片的复制，所以自己在找到脚本的基础上，补充了一部分内容。

功能：

• 将xml文件转换为JSON文件
• 将train/val/test中的图片复制

数据集准备：

• PASCAL VOC格式的数据

• 自己创建在根目录下的labels.txt：包含有数据集中出现的所有类别名称。
  
  放置在和Annotation同级的文件夹中：
  

代码运行

参数修改

141-143行提供以下三个参数来修改。

    # 存放数据集的根目录
    parser.add_argument('--root_dir', type=str, default='I:/VOC2007')
    # 数据集的图片类型，'jpg'或'png'
    parser.add_argument('--img_type', type=str, default='png')
    # 需要转换的数据集类型，'train', 'val', 'test'
    parser.add_argument('--data_type', type=str,default='test')

脚本代码

#!/usr/bin/python3
# -*- coding: utf-8 -*-
# @Time : 2023-03-08 11:27
#  script by dapeng:760715590@qq.com
#  PASCAl VOC 转换为 COCO 格式的脚本
import os
import argparse
import json
import xml.etree.ElementTree as ET
from typing import Dict, List
import re
import shutil


def get_label2id(labels_path: str) -> Dict[str, int]:
    """id is 1 start"""
    with open(labels_path, 'r') as f:
        labels_str = f.read().split()
    labels_ids = list(range(1, len(labels_str) + 1))
    return dict(zip(labels_str, labels_ids))


def get_annpaths(ann_dir_path: str = None,
                 ann_ids_path: str = None,
                 ext: str = '',
                 annpaths_list_path: str = None) -> List[str]:
    # If use annotation paths list
    if annpaths_list_path is not None:
        with open(annpaths_list_path, 'r'<