COCO数据集josn转txt格式

最新推荐文章于 2025-05-22 16:09:05 发布

困熊TrappedBear

最新推荐文章于 2025-05-22 16:09:05 发布

阅读量5.7k

点赞数 5

CC 4.0 BY-SA版权

分类专栏： New的小货柜文章标签： coco转txt格式

本文链接：https://blog.youkuaiyun.com/qq_40025335/article/details/90742646

1、coco_2104数据集

2、json格式转xml

3、xml转txt格式

思路：

把coco_train2014.json文件转换为xml;

coco_label-6.py：将xml格式文件转换txt文件保存在labels

得到文件labels，获取labels里文件名，生成文件名列表,从而生成训练集文件路径

代码实现

把coco_train2014.json文件转换为xml;并把保存对images

# -*- coding:utf-8 -*-
# coco-process.py
# http://www.manongjc.com/article/28607.html
from pycocotools.coco import COCO
# COCO转VOC
import os
import shutil
from tqdm import tqdm
import skimage.io as io
import matplotlib.pyplot as plt
import cv2
from PIL import Image, ImageDraw

#the path you want to save your results for coco to voc
savepath="/home/nxt/niuxiaotong/darknet/COCO/resultscar/"
img_dir=savepath+'images/'
anno_dir=savepath+'annotations/'
datasets_list=['train2014']
# datasets_list=['train2017']

classes_names = ['car', 'bus', 'truck']
#Store annotations and train2014/val2014/... in this folder
#dataDir= '/home/nxt/niuxiaotong/darknet/COCO'   # 换成json标注文件路径
dataDir= '/home/nxt/hyperai_data/au_dataset/darknet/coco'
headstr = """\
<annotation>
    <folder>VOC</folder>
    <filename>%s</filename>
    <source>
        <database>My Database</database>
        <annotation>COCO</annotation>
        <image>flickr</image>
        <flickrid>NULL</flickrid>
    </source>
    <owner>
        <flickrid>NULL</flickrid>
        <name>company</name>
    </owner>
    <size>
        <width>%d</width>
        <height>%d</height>
        <depth>%d</depth>
    </siz