Pascal voc 数据格式转 COCO 格式

最新推荐文章于 2024-08-16 20:00:00 发布

IT远征军

最新推荐文章于 2024-08-16 20:00:00 发布

阅读量1.2w

点赞数 13

分类专栏：目标检测算法

本文链接：https://blog.csdn.net/w113691/article/details/80817186

版权

本文介绍如何使用Python将PASCAL VOC格式的XML标注数据转换为COCO格式的JSON文件，包括解析XML，创建JSON结构，并保存转换后的数据。

摘要由CSDN通过智能技术生成

我们一般使用的数据格式是voc2007的数据格式，有时我们也需要用coco数据格式，这种格式一般以json格式存储，那么如将voc2007格式的数据转成coco数据格式呢？下面是python的实现：

# -*- coding:utf-8 -*-
# !/usr/bin/env python

import argparse
import json
import matplotlib.pyplot as plt
import skimage.io as io
import cv2
from labelme import utils
import numpy as np
import glob
import PIL.Image
import os,sys


class PascalVOC2coco(object):
    def __init__(self, xml=[], save_json_path='./new.json'):
        '''
        :param xml: 所有Pascal VOC的xml文件路径组成的列表
        :param save_json_path: json保存位置
        '''
        self.xml = xml
        self.save_json_path = save_json_path
        self.images = []
        self.categories = []
        self.annotations = []
        # self.data_coco = {}
        self.label = []
        self.annID = 1
        self.height = 0
        self.width = 0
        self.ob = []

        self.save_json()

    def data_transfer(self):
        for num, json_file in enumerate(self.xml):

            # 进度输出
            sys.stdout.write('\r>> Converting image %d/%d' % (
                num + 1, len(self.xml)))
            sys.stdout.flush()

            self.json_file = json_file
            #print("self.json", self.json_file)
            self.num = num 
            #print(self.num)
            path = os.path.dirname(self.json_file)
            #print(path)
            path = os.path.dirname(path)
            #print(path)
            # path=os.path.split(self.json_file)[0]
            # path=os.path.split(path)[0]
            obj_path = glob.glob(os.path.join(path, 'SegmentationObject', '*.png'))
            #print(obj_path)
            with open(json_file, 'r') as fp:
                #print(fp)
                flag = 0
                for p in fp:
                    #print(p)
                    # if 'folder' in p:
                    #     folder =p.split('>')[1].split('<')[0]
                    f_name = 1
                    if 'filename' in p:
                        self.filen_ame = p.split('>')[1].split('&l