|
import argparse |
|
import jsonlines |
|
from tqdm import tqdm |
|
import json |
|
from pycocotools.coco import COCO |
|
|
|
def dump_label_map(args): |
|
coco = COCO(args.input) |
|
cats = coco.loadCats(coco.getCatIds()) |
|
nms = {cat['id']-1:cat['name'] for cat in cats} |
|
with open(args.output,"w") as f: |
|
json.dump(nms, f) |
|
|
|
def coco_to_xyxy(bbox): |
|
x, y, width, height = bbox |
|
x1 = round(x, 2) |
|
y1 = round(y, 2) |
|
x2 = round(x + width, 2) |
|
y2 = round(y + height, 2) |
|
return [x1, y1, x2, y2] |
|
|
|
|
|
def coco2odvg(args): |
|
coco = COCO(args.input) |
|
cats = coco.loadCats(coco.getCatIds()) |
|
nms = {cat['id']:cat['name'] for cat in cats} |
|
metas = [] |
|
for img_id, img_info in tqdm(coco.imgs.items()): |
|
ann_ids = coco.getAnnIds(imgIds=img_id) |
|
instance_list = [] |
|
for ann_id in ann_ids: |
|
ann = coco.anns[ann_id] |
|
bbox = ann['bbox'] |
|
bbox_xyxy = coco_to_xyxy(bbox) |
|
label = ann['category_id'] |
|
category = nms[label] |
|
instance_list.append({ |
|
"bbox": bbox_xyxy, |
|
"label": label - 1, |
|
"category": category |
|
} |
|
) |
|
metas.append( |
|
{ |
|
"filename": img_info["file_name"], |
|
"height": img_info["height"], |
|
"width": img_info["width"], |
|
"detection": { |
|
"instances": instance_list |
|
} |
|
} |
|
) |
|
print(" == dump meta ...") |
|
with jsonlines.open(args.output, mode="w") as writer: |
|
writer.write_all(metas) |
|
print(" == done.") |
|
|
|
|
|
if __name__ == "__main__": |
|
parser = argparse.ArgumentParser("coco to odvg format.", add_help=True) |
|
parser.add_argument("--input", '-i', required=True, type=str, help="input list name") |
|
parser.add_argument("--output", '-o', required=True, type=str, help="output list name") |
|
parser.add_argument("--output_label_map", '-olm', action="store_true", help="output label map or not") |
|
args = parser.parse_args() |
|
|
|
if args.output_label_map: |
|
dump_label_map(args) |
|
else: |
|
coco2odvg(args) |
