Spaces:
Runtime error
Runtime error
File size: 3,474 Bytes
68896f3 |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 72 73 74 75 76 77 78 79 80 81 82 83 84 85 86 87 88 89 90 91 92 93 94 95 96 97 98 99 100 101 |
#encoding:utf-8
import os
import json
import cv2
from lxml import etree
import xml.etree.cElementTree as ET
import time
import pandas as pd
from tqdm import tqdm
import json
import argparse
# Run this command
# python coco2voc.py --json_path result.json
# This creates xml file in the image directory as specified in the json file
# images + json inside images folder
# converted xml file also in images folder
def coco2voc(anno, xml_dir, rgb_img):
with open(anno, 'r', encoding='utf-8') as load_f:
f = json.load(load_f)
imgs = f['images']
df_cate = pd.DataFrame(f['categories'])
_ = df_cate.sort_values(["id"],ascending=True)
df_anno = pd.DataFrame(f['annotations'])
categories = dict(zip(df_cate.id.values, df_cate.name.values))
for i in tqdm(range(len(imgs))):
xml_content = []
file_name = imgs[i]['file_name']
height = imgs[i]['height']
img_id = imgs[i]['id']
width = imgs[i]['width']
if(rgb_img == "True"): depth=3
else: depth=1
# print(depth)
xml_content.append("<annotation>")
xml_content.append(" <folder>VOC2007</folder>")
xml_content.append(" <filename>"+file_name+"</filename>")
xml_content.append(" <size>")
xml_content.append(" <width>"+str(width)+"</width>")
xml_content.append(" <height>"+str(height)+"</height>")
xml_content.append(" <depth>"+str(depth)+"</depth>")
xml_content.append(" </size>")
xml_content.append(" <segmented>0</segmented>")
annos = df_anno[df_anno["image_id"].isin([img_id])]
for index, row in annos.iterrows():
bbox = row["bbox"]
category_id = row["category_id"]
cate_name = categories[category_id]
xml_content.append(" <object>")
xml_content.append(" <name>"+cate_name+"</name>")
xml_content.append(" <pose>Unspecified</pose>")
xml_content.append(" <truncated>0</truncated>")
xml_content.append(" <difficult>0</difficult>")
xml_content.append(" <bndbox>")
xml_content.append(" <xmin>"+str(int(bbox[0]))+"</xmin>")
xml_content.append(" <ymin>"+str(int(bbox[1]))+"</ymin>")
xml_content.append(" <xmax>"+str(int(bbox[0]+bbox[2]))+"</xmax>")
xml_content.append(" <ymax>"+str(int(bbox[1]+bbox[3]))+"</ymax>")
xml_content.append(" </bndbox>")
xml_content.append(" </object>")
xml_content.append("</annotation>")
x = xml_content
xml_content=[x[i] for i in range(0,len(x)) if x[i]!="\n"]
xml_path = os.path.join(xml_dir,file_name.split('.')[-2] + '.xml')
with open(xml_path, 'w+',encoding="utf8") as f:
f.write('\n'.join(xml_content))
xml_content[:]=[]
if __name__ == '__main__':
parser = argparse.ArgumentParser(description="convert coco .json annotation to voc .xml annotation")
parser.add_argument('--json_path',type=str, help='path to json file.', default="./annotations/train.json")
parser.add_argument('--rgb_img', type=str, help='True if image in rgb', default="True")
parser.add_argument('--output',type = str, help='path to output xml files.',default="./images")
args = parser.parse_args()
if not os.path.exists(args.output):
os.mkdir(args.output)
coco2voc(args.json_path, args.output, args.rgb_img) |