labelme标注文件*.json转yolov5_v7.0sege 数据集文件*.txt

一.说明

yolov5 v7.0中添加了实例分割，实例分割所用到的数据集与目标检测所用到的数据集格式稍有不同。
yolov5 v6.0目标检测中的数据集文件*.txt中的格式为:

[类别ID] [目标框中心点P1_x] [目标框中心点P1_y]  [宽] [高]  
.
.
.
[类别ID] [目标框中心点Pn_x] [目标框中心点Pn_y]  [宽] [高]

每个目标框为归一化后的值，即每个框为：

类别ID 整数，从0开始
目标框中心点P1_x = 目标框P1中心点在图像中的X坐标 / 图像的宽
目标框中心点P1_y = 目标框P1中心点在图像中的Y坐标 / 图像的高
宽 = 目标框P1的宽 / 图像的宽
高 = 目标框P1的高 / 图像的高

---

yolov5 v7.0实例分割中的数据集文件*.txt中的格式为:

[类别ID] [轮廓点P1_x] [轮廓点P1_y]  [轮廓点P2_x] [轮廓点P2_y] ......  [轮廓点Pn_x] [轮廓点Pn_y]  
.
.
.
[类别ID] [轮廓点P1_x] [轮廓点P1_y]  [轮廓点P2_x] [轮廓点P2_y] ......  [轮廓点Pn_x] [轮廓点Pn_y]  

每个轮廓点为归一化后的值，即每个点为：

类别ID 整数，从0开始
轮廓点P1_x = 轮廓点P1在图像中的X坐标 / 图像的宽
轮廓点P1_y = 轮廓点P1在图像中的Y坐标 / 图像的高

二.代码

import json
import os
import glob
import os.path as osp


def labelme2yolov2Seg(jsonfilePath="", resultDirPath="", classList=["car", "human"]):
    """
    此函数用来将labelme软件标注好的数据集转换为yolov5_7.0sege中使用的数据集
    :param jsonfilePath: labelme标注好的*.json文件所在文件夹
    :param resultDirPath: 转换好后的*.txt保存文件夹
    :param classList: 数据集中的类别标签
    :return:
    """
    # 0.创建保存转换结果的文件夹
    if(not os.path.exists(resultDirPath)):
        os.mkdir(resultDirPath)

    # 1.获取目录下所有的labelme标注好的Json文件，存入列表中
    jsonfileList = glob.glob(osp.join(jsonfilePath, "*.json"))
    print(jsonfileList)  # 打印文件夹下的文件名称

    # 2.遍历json文件，进行转换
    for jsonfile in jsonfileList:
        # 3. 打开json文件
        with open(jsonfile, "r") as f:
            file_in = json.load(f)

            # 4. 读取文件中记录的所有标注目标
            shapes = file_in["shapes"]

            # 5. 使用图像名称创建一个txt文件，用来保存数据
            with open(resultDirPath + "\\" + jsonfile.split("\\")[-1].replace(".json", ".txt"), "w") as file_handle:
                # 6. 遍历shapes中的每个目标的轮廓
                for shape in shapes:
                    # 7.根据json中目标的类别标签，从classList中寻找类别的ID，然后写入txt文件中
                    file_handle.writelines(str(classList.index(shape["label"])) + " ")

                    # 8. 遍历shape轮廓中的每个点，每个点要进行图像尺寸的缩放，即x/width, y/height
                    for point in shape["points"]:
                        x = point[0]/file_in["imageWidth"]  # mask轮廓中一点的X坐标
                        y = point[1]/file_in["imageHeight"]  # mask轮廓中一点的Y坐标
                        file_handle.writelines(str(x) + " " + str(y) + " ")  # 写入mask轮廓点

                    # 9.每个物体一行数据，一个物体遍历完成后需要换行
                    file_handle.writelines("\n")
            # 10.所有物体都遍历完，需要关闭文件
            file_handle.close()
        # 10.所有物体都遍历完，需要关闭文件
        f.close()

if __name__ == "__main__":
    jsonfilePath = "Z:\\code\\PythonTest\\jsonTest\\jsons\\"  # 要转换的json文件所在目录
    resultDirPath = "Z:\\code\\PythonTest\\jsonTest\\results\\"  # 要生成的txt文件夹
    labelme2yolov2Seg(jsonfilePath=jsonfilePath, resultDirPath=resultDirPath, classList=["brick", "aggregate"])

三.运行结果

原始*.json文件所在目录

生成的*.txt所在目录

转换生成的*.txt文件内容