#?-*-?coding:?utf-8?-*-
“““
Created?on?Thu?Jul?11?15:34:57?2019

@author:?Kevin
“““

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
?
#the?path?you?want?to?save?your?results?for?coco?to?voc
savepath=“F:/coco2014/result/“
img_dir=savepath+‘images/‘
anno_dir=savepath+‘Annotations/‘
#?datasets_list=[‘train2014‘?‘‘]
datasets_list=[‘val2014‘]
?
classes_names?=?[‘person‘?‘bicycle‘?‘car‘?‘motorcycle‘?‘a(chǎn)irplane‘?‘bus‘?‘train‘?‘truck‘?‘boat‘?‘traffic?light‘?‘fire?hydrant‘?‘stop?sign‘?‘parking?meter‘?‘bench‘?‘bird‘?‘cat‘?‘dog‘?‘horse‘?‘sheep‘?‘cow‘?‘elephant‘?‘bear‘?‘zebra‘?‘giraffe‘?‘backpack‘?‘umbrella‘?‘handbag‘?‘tie‘?‘suitcase‘?‘frisbee‘?‘skis‘?‘snowboard‘?‘sports?ball‘?‘kite‘?‘baseball?bat‘?‘baseball?glove‘?‘skateboard‘?‘surfboard‘?‘tennis?racket‘?‘bottle‘?‘wine?glass‘?‘cup‘?‘fork‘?‘knife‘?‘spoon‘?‘bowl‘?‘banana‘?‘a(chǎn)pple‘?‘sandwich‘?‘orange‘?‘broccoli‘?‘carrot‘?‘hot?dog‘?‘pizza‘?‘donut‘?‘cake‘?‘chair‘?‘couch‘?‘potted?plant‘?‘bed‘?‘dining?table‘?‘toilet‘?‘tv‘?‘laptop‘?‘mouse‘?‘remote‘?‘keyboard‘?‘cell?phone‘?‘microwave‘?‘oven‘?‘toaster‘?‘sink‘?‘refrigerator‘?‘book‘?‘clock‘?‘vase‘?‘scissors‘?‘teddy?bear‘?‘hair?drier‘?‘toothbrush‘]
#Store?annotations?and?train2014/val2014/...?in?this?folder
dataDir=?‘F:/coco2014/‘
?
headstr?=?“““\

????VOC
????%s
????
????????se>My?Databasese>
????????COCO
????????flickr
????????NULL
????
????
????????NULL
????????company
????
????
????????%d
????????%d
????????%d
????
????0
“““
objstr?=?“““\
????ject>
????????%s
????????Unspecified
????????0
????????0
????????
????????????%d
????????????%d
????????????%d
????????????%d
????????
????ject>
“““
?
tailstr?=?‘‘‘\

‘‘‘
?
#if?the?dir?is?not?existsmake?itelse?delete?it
def?mkr（path）:
????if?os.path.exists（path）:
????????shutil.rmtree（path）
????????os.mkdir（path）
????else:
????????os.mkdir（path）
mkr（img_dir）
mkr（anno_dir）
def?id2name（coco）:
????classes=dict（）
????for?cls?in?coco.dataset[‘categories‘]:
????????classes[cls[‘id‘]]=cls[‘name‘]
????return?classes
?
def?write_xml（anno_pathhead?objs?tail）:
????f?=?open（anno_path?“w“）
????f.write（head）
????for?obj?in?objs:
????????f.write（objstr%（obj[0]ob