coco数据集提取特定的类转化成voc数据集

最新推荐文章于 2025-07-10 09:44:41 发布

原创

最新推荐文章于 2025-07-10 09:44:41 发布 · 756 阅读

7 ·

CC 4.0 BY-SA版权

文章标签：

#python

这段代码是一个Python脚本，用于从COCO数据集中提取特定类（如person），将图像和对应的注释转换为VOC格式的XML文件。它使用COCOAPI处理数据，包括读取COCOJSON文件，生成XML头部和对象标签，以及保存图片和XML文件。

coco_split_voc.py

from pycocotools.coco import COCO
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
 
# 需要设置的路径
savepath="/path/to/generate/COCO/" 
img_dir=savepath+'images/'
anno_dir=savepath+'annotations/'
datasets_list=['train2017', 'val2017']

#coco有80类，这里写要提取类的名字，以person为例 
classes_names = ['person'] 
#包含所有类别的原coco数据集路径
'''
目录格式如下：
$COCO_PATH
----|annotations
----|train2017
----|val2017
----|test2017
'''
dataDir= '/path/to/coco_orgi/' 
 
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>