File size: 3,385 Bytes
77b1729 |
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 102 103 104 |
import json
import os
# Function to convert a single JSON file to COCO format
def convert_to_coco(json_data):
# Initialize COCO data structure
coco_data = {
"info": {},
"licenses": [],
"images": [],
"annotations": [],
"categories": []
}
# Populate 'info' field (optional)
coco_data["info"] = {
"description": "Converted from custom format to COCO format",
"url": "http://cocodataset.org",
"version": "1.0",
"year": 2024,
"contributor": "Anonymous",
"date_created": "2024-06-29"
}
# Populate 'images' field
image_data = {
"id": 1,
"file_name": json_data["filename"],
"width": int(json_data["size"]["width"]),
"height": int(json_data["size"]["height"]),
"license": None,
"flickr_url": None,
"coco_url": None,
"date_captured": None
}
coco_data["images"].append(image_data)
# Populate 'annotations' field
annotation_id = 1
for obj in json_data["object"]:
annotation = {
"id": annotation_id,
"image_id": image_data["id"],
"category_id": None, # This needs to be mapped to COCO categories
"segmentation": [],
"area": 0,
"bbox": [
float(obj["bndbox"]["xmin"]),
float(obj["bndbox"]["ymin"]),
float(obj["bndbox"]["xmax"]) - float(obj["bndbox"]["xmin"]),
float(obj["bndbox"]["ymax"]) - float(obj["bndbox"]["ymin"])
],
"iscrowd": 0,
"attributes": obj.get("attributes", [])
}
coco_data["annotations"].append(annotation)
annotation_id += 1
return coco_data
# Directory containing JSON files
json_dir = "/path/to/your/json/files/"
# Initialize a list to hold all COCO data dictionaries
all_coco_data = []
# Iterate over each JSON file in the directory
for filename in os.listdir(json_dir):
if filename.endswith(".json"):
json_path = os.path.join(json_dir, filename)
with open(json_path, 'r') as f:
json_data = json.load(f)
# Convert JSON data to COCO format
coco_data = convert_to_coco(json_data)
# Append to the list of all COCO data
all_coco_data.append(coco_data)
# Optionally, merge all COCO data into one dictionary
merged_coco_data = {
"info": all_coco_data[0]["info"], # Assuming info is the same across all files
"licenses": all_coco_data[0]["licenses"], # Assuming licenses are the same across all files
"images": [],
"annotations": [],
"categories": all_coco_data[0]["categories"] # Assuming categories are the same across all files
}
# Merge images and annotations
for coco_data in all_coco_data:
merged_coco_data["images"].extend(coco_data["images"])
merged_coco_data["annotations"].extend(coco_data["annotations"])
# Convert to JSON format
merged_json = json.dumps(merged_coco_data, indent=4)
print(merged_json)
# Optionally, save merged JSON to a file
output_path = os.path.join(json_dir, "merged_coco_data.json")
with open(output_path, 'w') as f:
json.dump(merged_coco_data, f, indent=4)
print(f"Merged COCO data saved to {output_path}")
|