Spaces:
Sleeping
Sleeping
File size: 1,014 Bytes
0d8d0b4 | 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 | # https://github.com/chrise96/image-to-coco-json-converter/tree/master
from PIL import Image
import numpy as np
from skimage import measure
import os
import json
def create_category_annotation(category_dict):
"""
Create category annotations in COCO JSON format.
Args:
category_dict (dict): Dictionary containing category names and IDs.
Returns:
list: List of category annotations.
"""
category_list = []
for key, value in category_dict.items():
category = {
"supercategory": key,
"id": value,
"name": key
}
category_list.append(category)
return category_list
def get_coco_json_format():
"""
Get the standard COCO JSON format.
Returns:
dict: COCO JSON format skeleton.
"""
# Standard COCO format
coco_format = {
"info": {},
"licenses": [],
"images": [{}],
"categories": [{}],
"annotations": [{}]
}
return coco_format
|