python实现提取COCO,VOC数据集中特定的类

python实现提取实现提取COCO,VOC数据集中特定的类数据集中特定的类

主要介绍了python实现提取COCO,VOC数据集中特定的类，具有很好的参考价值，希望对大家有所帮助。一起

跟随小编过来看看吧

1.python提取提取COCO数据集中特定的类数据集中特定的类

安装pycocotools github地址：https://github.com/philferriere/cocoapi

pip install git+https://github.com/philferriere/cocoapi.git#subdirectory=PythonAPI

提取特定的类别如下：

from pycocotools.coco import COCO

from PIL import Image, ImageDraw

#the path you want to save your results for coco to voc

savepath="/media/huanglong/Newsmy/COCO/" #保存提取类的路径,我放在同一路径下

img_dir=savepath+'images/'

anno_dir=savepath+'Annotations/'

# datasets_list=['train2014', 'val2014']

datasets_list=['train2014']

classes_names = ['person'] #coco有80类，这里写要提取类的名字，以person为例

<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>

<segmented>0</segmented>

"""

objstr = """\

<object>

<name>%s</name>

<pose>Unspecified</pose>

<truncated>0</truncated>

<difficult>0</difficult>

<bndbox>

<xmin>%d</xmin>

tailstr = '''\

</annotation>

'''

#if the dir is not exists,make it,else delete it

def mkr(path):

if os.path.exists(path):

shutil.rmtree(path)

os.mkdir(path)