YOLO:将json文件转换为txt格式

因为YOLO训练的时候有自己的标注格式,因此需要将labelme标注的json格式的数据转成yolo需要的txt格式。
labelme的标注以及json各部分的含义在上一篇中已经详细介绍。
今天记录的是如何将json转换成txt。
yolo需要的标注是五个值:类别 中心点x坐标 中心点y坐标 框的宽 框的高。
并将这五个值按行存在txt文件中。
因此我们需要将json中相应的值进行转换,并写入txt中。
1、将左上角右下角坐标注转换为中心点坐标
注:在转换时同时需要做归一化,归一化的方法是除以图片的宽或高,以得到0~1之间的值。
代码如下:

*#将json的坐标转换为yolo的坐标转换:x,y为中心点坐标,w h为框的宽和高
def convert(img_size, box):  # 坐标转换
    dw = 1

你可能感兴趣的:(json,YOLO,python)