转换为txt格式"/>
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. / (img_size[0]) dh = 1. / (img_size[1])x = (box[0] + box[2]) / 2.0 * dwy = (box[1] + box[3]) / 2.0 * dhw = (box[2] - box[0]
更多推荐
YOLO:将json文件转换为txt格式
发布评论