在Python中将CSV表转换为COCO格式,可以通过以下步骤实现:
import csv
import json
csv_file = open('input.csv', 'r')
csv_reader = csv.reader(csv_file)
header = next(csv_reader) # 获取CSV文件的表头
data = []
for row in csv_reader:
data.append(row)
coco_data = {
"info": {},
"licenses": [],
"categories": [],
"images": [],
"annotations": []
}
image_id = 1
annotation_id = 1
for row in data:
image = {
"id": image_id,
"file_name": row[0], # 图像文件名
"width": int(row[1]), # 图像宽度
"height": int(row[2]) # 图像高度
}
coco_data["images"].append(image)
annotation = {
"id": annotation_id,
"image_id": image_id,
"category_id": int(row[3]), # 目标类别ID
"bbox": [int(row[4]), int(row[5]), int(row[6]), int(row[7])] # 边界框坐标
}
coco_data["annotations"].append(annotation)
image_id += 1
annotation_id += 1
with open('output.json', 'w') as json_file:
json.dump(coco_data, json_file)
以上代码将CSV表转换为COCO格式的JSON文件。在实际应用中,可以根据需要进行适当的修改和扩展。
腾讯云相关产品和产品介绍链接地址:
领取专属 10元无门槛券
手把手带您无忧上云