txt转为coco数据集

将txt格式的数据集转换为coco格式的数据集可以通过以下步骤实现：

# 导入必要的库
import json
import os

# 定义函数，将txt格式的标注转换为coco格式的标注
def txt2coco(txt_path, img_dir):
    # 定义coco格式的数据结构
    coco = {
        "images": [],
        "annotations": [],
        "categories": []
    }
    # 定义类别列表
    classes = []
    # 读取txt文件
    with open(txt_path, "r") as f:
        lines = f.readlines()
    # 遍历每一行
    for line in lines:
        # 获取图片路径和标注信息
        img_path, *labels = line.strip().split()
        # 获取图片信息
        img_name = os.path.basename(img_path)
        img_id = int(os.path.splitext(img_name)[0])
        img = {
            "id": img_id,
            "file_name": img_name
        }
        coco["images"].append(img)
        # 遍历每个标注信息
        for label in labels:
            # 获取类别和标注框信息
            class_name, x, y, w, h = label.split(",")
            # 添加类别到类别列表
            if class_name not in classes:
                classes.append(class_name)
                coco["categories"].append({
                    "id": len(classes),
                    "name": class_name
                })
            # 添加标注信息到标注列表
            annotation = {
                "id": len(coco["annotations"]) + 1,
                "image_id": img_id,
                "category_id": classes.index(class_name) + 1,
                "bbox": [int(x), int(y), int(w), int(h)],
                "area": int(w) * int(h),
                "iscrowd": 0
            }
            coco["annotations"].append(annotation)
    return coco

# 调用函数，将txt格式的标注转换为coco格式的标注
coco_data = txt2coco("annotations.txt", "images/")

# 将coco格式的标注保存为json文件
with open("coco_annotations.json", "w") as f:
    json.dump(coco_data, f)

相关问题：